dataframe取特定几列组成新的dataframe
时间: 2023-11-18 08:00:20 浏览: 103
python DataFrame 取差集实例
你可以使用 Pandas 库来取特定几列组成新的 dataframe。下面是一个示例代码:
```python
import pandas as pd
# 创建一个示例 dataframe
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'Gender': ['Female', 'Male', 'Male', 'Male'],
'Salary': [5000, 6000, 7000, 8000]
})
# 取出 Name、Age 和 Salary 列组成新的 dataframe
new_df = df[['Name', 'Age', 'Salary']]
print(new_df)
```
输出结果为:
```
Name Age Salary
0 Alice 25 5000
1 Bob 30 6000
2 Charlie 35 7000
3 David 40 8000
```
在这个示例中,我们首先创建了一个包含 Name、Age、Gender 和 Salary 列的 dataframe。然后,我们使用 `df[['Name', 'Age', 'Salary']]` 来取出 Name、Age 和 Salary 列组成新的 dataframe。这里使用双重方括号是因为我们需要传递一个列名的列表给 dataframe。
阅读全文