如何在Python中的函数中创建一个新的空白数组



我有一个函数可以从父数组中提取一列数据,但我不知道如何让该函数为要存储的数据创建一个新的空白数组。

该函数有两个变量";标题";以及";列";,其中";标题";将是所提取数据的新数组的期望名称(当前我只是输入预定义数组的名称(;列";定义要提取父数组中的哪一列。

我还应该声明我使用的是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([])

最新更新