【pandas】DataFrame获取特定的列

问题需求:提取所有列名中包含“线索”、“浏览”字段的列

import pandas as pd

path = 'F:\python_projects\python_learning\ershouche.csv'
df = pd.read_csv(open(path), index_col=0)
df = df.fillna(0)  # 填充空值后需赋值
print(df.describe())

columns = df.columns.values.tolist()  # 获取列名列表,注意values,tolist的使用
col_xian = []  # 存储包含‘线索’字段的列名
for i in columns:
    if '线索' in i:
        col_xian.append(i)

col_liu = [] # 存储包含‘浏览’字段的列名
for i in columns:
    if '浏览' in i:
        col_liu.append(i)

df_xian = df[col_xian]  # 根据列名取列
df_liu = df[col_liu]

更多信息请参考http://pandas.pydata.org/pandas-docs/stable/10min.html#selection

你可能感兴趣的:(pandas)