python - 迭代 for 循环并应用 np.union1d

Question

我想做np.union1d通过迭代 for 循环。这是我正在使用的代码:

arr = np.empty((1,), dtype=np.int32)
for i in range(2):
    arr = np.union1d(arr, data[df.iloc[i,7]])

data是一本字典，我想根据数据框 df 中定义的键从中提取值。抱歉，我无法为您提供 data 的更多详细信息。 & df因为商业 secret 。运行此循环后，我看到 arr 显示空数组，而 data[df.iloc[i,7]]每次迭代都会生成大约 10K 大小的数组。你能让我知道我做错了什么吗？

Answer 1

为了确定，请检查您的代码实际上不是这样的:

for i in range(2):
    arr=np.empty((1,), dtype=np.int32)
    arr=np.union1d(arr, data[df.iloc[i,7]])

如果没有，我会按以下方式调试

for i in range(2):
    new_arr = data[df.iloc[i,7]]
    print(type(new_arr))  # check that it's an array
    print(new_arr.shape)  # check that first dim > 1, second dim = 1

一旦你开始工作，我还建议初始化arr = []，因为np.empty通过放置一个非常非常小的值作为占位符来初始化数组，所以它将出现在您的最终连接数组中!