dataframe提取其中某几列构成新的表
时间: 2023-11-18 20:13:03 浏览: 130
pandas 选择某几列的方法
可以使用 Pandas 的 `DataFrame` 中的 `loc` 或者 `iloc` 方法来提取某几列构成新的表。下面是一个示例代码:
```python
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({
'name': ['Alice', 'Bob', 'Charlie'],
'age': [25, 30, 35],
'gender': ['F', 'M', 'M'],
'salary': [5000, 6000, 7000]
})
# 提取 name, gender, salary 三列构成新的表
new_df = df.loc[:, ['name', 'gender', 'salary']]
# 或者使用 iloc 方法
# new_df = df.iloc[:, [0, 2, 3]]
print(new_df)
```
输出结果为:
```
name gender salary
0 Alice F 5000
1 Bob M 6000
2 Charlie M 7000
```
其中 `df.loc[:, ['name', 'gender', 'salary']]` 表示提取所有行的 `name`, `gender`, `salary` 三列,`df.iloc[:, [0, 2, 3]]` 表示提取所有行的第 0, 2, 3 列。
阅读全文