list(df)方法可以直接获得df的列名
import numpy as np
import pandas as pd
df = pd.DataFrame(data=[{"A":"one", "B":1, "C":11}, {"A":"two", "B":2, "C":22}])
A | B | C | |
---|---|---|---|
0 | one | 1 | 11 |
1 | two | 2 | 22 |
columns_list = list(df)
print(columns_list)
["A","B","C"]
columns_list = str(list(df)).replace("[", "(").replace("]", ")")
print(columns_list)
("A","B","C")
一般来说可以用在选择数据 - df[][] - df[选择指定列][筛选条件]