如何在pandas中提取表格的列
时间: 2024-05-15 09:19:18 浏览: 103
Python基于pandas爬取网页表格数据
5星 · 资源好评率100%
可以使用pandas的loc或iloc方法来提取表格的列。loc方法通过列名来定位列,iloc方法通过列的索引号来定位列。
例如,假设有一个名为df的数据框,其中有三列a、b、c,可以使用以下代码提取列a和b:
```
import pandas as pd
# 创建数据框
data = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}
df = pd.DataFrame(data)
# 提取列a和b
df_ab = df.loc[:, ['a', 'b']]
df_ab = df.iloc[:, [0, 1]]
```
在上面的代码中,df.loc[:, ['a', 'b']]表示提取所有行的列a和b,df.iloc[:, [0, 1]]表示提取所有行的第0列和第1列(即列a和b)。
如果想要提取单独一列,可以使用以下代码:
```
# 提取列a
df_a = df.loc[:, 'a']
df_a = df.iloc[:, 0]
```
在上面的代码中,df.loc[:, 'a']表示提取所有行的列a,df.iloc[:, 0]表示提取所有行的第0列(即列a)。
阅读全文