pycharm 读取dataframe 某几列数据
时间: 2023-09-23 14:07:21 浏览: 84
pycharm连接autodl服务器(yolov8训练自己的数据集)
可以使用dataframe的loc或iloc方法选择需要的列。
loc方法使用列名作为索引,例如:
```python
import pandas as pd
df = pd.read_csv('data.csv')
# 选择第一列和第三列
df = df.loc[:, ['col1', 'col3']]
```
iloc方法使用列的位置作为索引,例如:
```python
import pandas as pd
df = pd.read_csv('data.csv')
# 选择第一列和第三列
df = df.iloc[:, [0, 2]]
```
以上代码分别选择了dataframe的第一列和第三列作为新的dataframe。
阅读全文