dateframe保留等于某个list的列名
时间: 2023-08-29 15:06:29 浏览: 37
你可以使用 Pandas 库中的 `isin()` 方法来保留等于某个列表的列名。以下是一个示例代码:
```python
import pandas as pd
# 创建一个示例 DataFrame
data = {'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 定义一个列表,包含你想要保留的列名
columns_to_keep = ['A', 'C']
# 使用 isin() 方法筛选 DataFrame,并保留指定的列名
df_filtered = df[df.columns[df.columns.isin(columns_to_keep)]]
# 打印筛选后的 DataFrame
print(df_filtered)
```
运行这段代码后,你会得到一个只包含列 "A" 和 "C" 的新 DataFrame。
相关问题
dataframe 获得某一列列名
要获取 DataFrame 中某一列的列名,可以使用 `columns` 属性来获取所有列名,然后根据列名来获取相应的列。例如,假设我们有一个 DataFrame `df`,其中包含 `name`、`age` 和 `gender` 三列数据,要获取 `name` 列的列名,可以按照以下方式操作:
```
import pandas as pd
# 创建一个示例 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie'],
'age': [25, 30, 35],
'gender': ['Female', 'Male', 'Male']}
df = pd.DataFrame(data)
# 获取列名列表
column_names = df.columns.tolist()
# 获取 name 列的列名
name_column_name = column_names[0]
print(name_column_name)
```
输出结果为:
```
name
```
这样就可以获取 DataFrame 中任意一列的列名了。
dataframe重命名第二列列名
你可以使用 pandas 中的 rename 方法来重命名 DataFrame 的列名,例如:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df = df.rename(columns={'B': 'new_name'})
print(df)
```
这样就可以将 DataFrame 的第二列列名从 'B' 改为 'new_name'。