我不知道这里出了什么问题。
d = {'x' : [1,4,6,9],
'y' : [1,4,6,8]}
df = pd.DataFrame(d)
#filter columns based on value in specific row
df_VIP = df.iloc[:,df.iloc[1:2,:]<3]
我得到了错误。我的真实数据框也会发生这种情况......
ValueError: 缓冲区的维数错误（预期为 1，得到 2）
我希望任何人都有解决它的提示。谢谢！
【问题讨论】：
标签： python-3.x pandas valueerror
如果可能，按一行选择，例如第二个由1 转换为numpy 数组，因为使用了iloc：
d = {'x' : [1,4,6,9],
'y' : [1,2,6,8]}
df = pd.DataFrame(d)
df_VIP = df.iloc[:,df.iloc[1,:].to_numpy()<3]
print (df_VIP)
y
0 1
1 2
2 6
3 8
如果使用您的选择 - 一行 DataFrame 而不是 Series 通过索引来展平值：
print (df.iloc[1:2,:])
x y
1 4 2
print (df.iloc[1:2,:].to_numpy())
[[4 2]]
print (df.iloc[1:2,:].to_numpy()[0])
[4 2]
df_VIP = df.iloc[:,df.iloc[1:2,:].to_numpy()[0]<3]
【讨论】：