python - pandas:to_csv 具有命名列的数字范围?

标签 python pandas

即使标题标有字符串,是否可以通过 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/

相关文章:

python - 在 Raspberry PI 上安装 "ring.cx SIP client"

python - 如何验证pandas.read_csv读取的csv数据?

python - 将稀疏数据帧转换为密集数据帧

python - 如何将模糊函数与 apply(lambda x : ) function?

python - pandas:read_csv仅排除某些行

python - 将多个自定义类与 Pipeline sklearn (Python) 结合使用

python - 无法使用anaconda安装ggp​​lot

python - ipython笔记本清除所有代码

python - User.py异常类型: TypeError Exception Value: 'Manager' object is not iterable

python - 检查目录是否包含具有给定扩展名的文件