列名包含数据处理.py 739 B

12345678910111213141516171819202122
  1. import pandas as pd
  2. path = r'd://data//11.csv'
  3. df = pd.read_csv(path, encoding='gb18030')
  4. df.reset_index(inplace=True)
  5. print(df.columns)
  6. df.columns = [i.replace('()', '') for i in df.columns]
  7. wind_names = set([i.split('#-')[0] for i in df.columns if i.find('#-') > -1])
  8. print(df.columns)
  9. print(wind_names)
  10. for wind_name in wind_names:
  11. select_cols = [i for i in df.columns if str(i).startswith(wind_name)]
  12. print(select_cols)
  13. select_cols.insert(0, '时间')
  14. print(select_cols)
  15. df_temp = df[select_cols]
  16. df_temp.columns = [i.split('#-')[-1] for i in df_temp.columns]
  17. df_temp.sort_values(by='时间', inplace=True)
  18. df_temp.to_csv("d://data//najiade//"+str(wind_name) + '.csv', encoding='utf8', index=False)