python - 在 Pandas iloc 中同时传递列索引和范围

Question

假设我有一个包含 100 列的数据框。我只想为所有行选择 0,1 和 51-100 列。我现在正在尝试这样的事情:

df.iloc[:,[0,1,range(51,101)]]

但它抛出错误:

“用序列设置数组元素”

尽管如果我只传递范围而不传递列索引，它就可以工作。

我也尝试过:

df.iloc[:,[0,1,51:102)]]

它显示无效语法。

有人可以指出这里的错误吗？另请建议实现功能的替代方法。

Answer 1

我认为你需要numpy.r_对于连接索引:

df.iloc[:,np.r_[0,1,51:102]]

示例:

df = pd.DataFrame([[0] * 10], columns=list('abcdefghij'))
print (df)
   a  b  c  d  e  f  g  h  i  j
0  0  0  0  0  0  0  0  0  0  0

print (df.iloc[:,np.r_[0,1,5:8]])
   a  b  f  g  h
0  0  0  0  0  0