pandas去掉column保存呢
时间: 2023-07-13 13:27:20 浏览: 74
十分钟学习Pandas
如果你想在保存 DataFrame 时去掉某些列,可以使用 `drop` 方法对这些列进行删除操作,然后再使用 `to_csv` 方法进行保存。
例如,假设你有一个名为 `df` 的 DataFrame,其中包含三列,你想要删除第二列,并将剩下的两列保存为 CSV 文件,可以按照以下方式操作:
```python
import pandas as pd
# 创建 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 删除 'B' 列
df = df.drop('B', axis=1)
# 保存为 CSV 文件
df.to_csv('data.csv', index=False)
```
在上面的代码中,`df.drop('B', axis=1)` 表示删除 `df` DataFrame 中的 'B' 列。`axis=1` 表示删除列,而不是行。然后,使用 `to_csv` 方法将剩下的两列保存为 CSV 文件。注意,`index=False` 用于不保存行索引。
阅读全文