取dataframe数据中第1列、第4列、第5列的所有数据
时间: 2024-05-06 10:21:05 浏览: 64
SQL语句,取出一个数据表的所有列名
你好,很高兴为您解答问题。针对您的问题,可以使用 pandas 中的 iloc 函数来实现。具体的代码如下:
```python
import pandas as pd
# 读取数据
df = pd.read_csv('data.csv')
# 取指定列的数据
result = df.iloc[:, [0, 3, 4]].values
print(result)
```
其中,`iloc` 函数用于在整个 DataFrame 中,通过下标索引的方式来获取指定的行和列。`[:, [0, 3, 4]]` 表示获取所有行,第 1 列、第 4 列、第 5 列的数据。`.values` 表示将获取的数据转换成矩阵形式输出。
希望能帮到您,如果还有其他问题,请继续问我哦。
阅读全文