即使标题标有字符串,是否可以通过 pd.to_csv 为 columns 参数提供数字范围?
示例数据框:
January February March April May June July August September
0 67 43 48 58 82 102 118 114 82
1 45 27 16 12 65 89 112 100 35
如果我想写出前八列,我必须使用:
pd.to_csv('<filename>',usecols=['January, February, March, April, May, June, July, August'])
这显然会占用大量空间,并且对于具有一百多个列的数据框来说是不切实际的。但我似乎无法使用一系列列或切片作为参数:
df.to_csv('filename.csv',columns=list(range(1,9)))
这将返回 KeyError: '[[1, 2, 3, 4, 5, 6, 7, 8]] 都不在 [列] 中'
。
我确信有一个简单的方法可以做到这一点,我只是找不到任何类似的问题......
最佳答案
您可以将数据帧的列名称作为一个系列访问,然后对其进行切片,例如:
df.to_csv('filename.csv', usecols=df.columns[:8])
关于python - pandas:to_csv 具有命名列的数字范围?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38351736/