我有一个函数可以从父数组中提取一列数据,但我不知道如何让该函数为要存储的数据创建一个新的空白数组。
该函数有两个变量";标题";以及";列";,其中";标题";将是所提取数据的新数组的期望名称(当前我只是输入预定义数组的名称(;列";定义要提取父数组中的哪一列。
我还应该声明我使用的是Jupyter笔记本。
这是代码:
def extract_col(title, column):
title = [data[:,column]]
其中";数据";代码中只有父数组的名称。这里有一个例子:
extract_col(array1, 3)
这个输入应该创建一个名为array1的数组,它由父数组的第3列"组成;数据";。
您可以使用这个来实现目标(这会创建一个您想要的变量(:
def extract_col(title, column):
locals()[title] = data[:,column]
但是,您不应该使用locals()
为此(或永远(创建新的变量。相反,使用字典,而不是变量:
def extract_col(title, column):
d = {}
d[title] = data[:,column]
然后你可以在需要的地方打电话给d[标题]。
如果你正在使用np.array()
,并且想要一个新的空数组,正如你在问题中提到的,你可以使用:
def extract_col(title, column):
d = {}
d[title] = np.array([])