问题需求:提取所有列名中包含“线索”、“浏览”字段的列
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