python取dataframe特定的列
时间: 2023-04-26 08:01:59 浏览: 479
Pandas库之DataFrame使用的学习笔记
要取DataFrame中特定的列,可以使用DataFrame的loc或iloc方法。其中,loc方法是基于标签的索引,而iloc方法是基于位置的索引。
例如,假设有一个DataFrame df,其中包含三列A、B、C,要取出B列,可以使用以下代码:
```
# 使用loc方法
df.loc[:, 'B']
# 使用iloc方法
df.iloc[:, 1]
```
其中,冒号表示取所有行,'B'或1表示取第二列。如果要取多列,可以将列名或位置放在一个列表中,如下所示:
```
# 取B和C两列
df.loc[:, ['B', 'C']]
# 取第二和第三列
df.iloc[:, [1, 2]]
```
阅读全文