Question

我想基于几个变量列表对我的数据框进行子集化，即：

 list1=[var1,var2,var3]
 list2=[var4,var5,var6]

 data_final = data[list1,list2]

产生此错误：

 TypeError: unhashable type: 'list'

如果我只提供一个列表，则一切正常：

 data_final = data[list1]

下面是一个最小示例：

dict1 = [{'var0': 0, 'var1': 1, 'var2': 2},
     {'var0': 0, 'var1': 2, 'var2': 4},
    {'var0': 1, 'var1': 5, 'var2': 8},
    {'var0': 1, 'var1': 15, 'var2': 12},]
 df = pd.DataFrame(dict1, index=['s1', 's2','s3','s4'])

  list1=['var0']
  list2=['var1','var2']

这两个命令可以正常工作：

  df[list1]
  df[list2]

但这会产生上述错误：

  df[list1,list2]

Answer 1

您需要将列名写在一个列表中，而不是列表中：

data_final= data[[var1,var2,var3],[var4,var5,var6]]

来自文档：

您可以将列列表传递到[]以按此顺序选择列。如果DataFrame中不包含列，则将引发异常。也可以通过这种方式设置多列

Answer 2

这是您期望的输出吗？

df[list1 + list2]
Out[106]: 
    var0  var1  var2
s1     0     1     2
s2     0     2     4
s3     1     5     8
s4     1    15    12

Answer 3

要将任意数量的列表加载到行的数据框中（只要列表的长度相等），您可以执行以下操作：

import pandas as pd
l1 = [1,2,3]
l2 = [10,20,30]
col_name = ['c1','c2','c3']
row_name = ['r1','r2']
pd.DataFrame([l1,l2],columns=col_name, index=row_name)

    c1  c2  c3
r1   1   2   3
r2  10  20  30

要将任意数量的列表加载到列的数据框中，您必须将列表压缩在一起，如下所示：

l1 = [1,2,3]
l2 = [10,20,30]
col_name = ['c1','c2']
row_name = ['r1','r2','r3']
zipped_list = list(zip(l1,l2))

import pandas as pd

pd.DataFrame(zipped_list,columns=col_name,index=row_name)

    c1  c2
r1   1  10
r2   2  20
r3   3  30

希望能帮助您，py-on！

带有列表的子集熊猫数据框

3 个答案: