python - 按列名称为数据框的每一列创建一个数组

Question

我正在尝试创建多个与大型数据框中每列的数据值相对应的数组，并且我希望每个数组与其对应的列共享相同的名称。以下是我尝试解决此问题的示例:

x = [a b c] <- column names
    [1 2 3]
    [1 2 3]

名称 = [a b c]

名称中的 i:

i = numpy.array(x[i])

我想创建三个数组(a、b 和 c):

a = [1, 1]
b = [2, 2]
c = [3, 3]

但我得到的只是一个数组:

i = [3, 3]

如何解决这个问题？

Answer 1

每次在 for 循环中，您都会为同一个变量分配不同的值，因此当循环结束时，您将保留最后分配的值。

我建议您创建一个字典并使用每个名称作为键，使用数组作为值:

arrays = {}
for i in names:
    arrays[i] = numpy.array(x[i])

然后您可以使用以下方法检索每个数组:

arrays[name]