请考虑这个 df:
df = pd.DataFrame({'a':[1,2], 'b':[1,2], 'c':[1,2], 'd':[1,2], 'e':[1,2], 'f':[1,2], 'g':[1,2], 'h':[1,2]})
a b c d e f g h
0 1 1 1 1 1 1 1 1
1 2 2 2 2 2 2 2 2
如何选择第 1、4 和 5-7 列? 我尝试了什么:
df.iloc[:, [0, 3, np.arange(5,8)]]
ValueError: setting an array element with a sequence.
最佳答案
你可以这样做:
df.iloc[:, [0, 3] + list(range(5,8))]
[0, 3] + list(range(5,8))
连接 2 个列表,将您的显式列表与从您所需范围派生的列表相结合。
或者,您可以使用 numpy.r
为您构建索引数组:
import numpy as np
df.iloc[:, np.r_[0,3,5:8]]
np.r_[0,3,5:8] # array([0, 3, 5, 6, 7])
这会很有用,例如,如果您有多个范围。
关于python - 如何选择结合了列表和范围的数据框列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49343637/