python csvファイル列抽出繰り返し構文について

Question

201908_14_14101010.csv　というファイルから
"測定局コード","日付","時","WD(16Dir)","WS(m/s)","TEMP(℃)"　の列のみ抽出し
14101010.csvという新しいファイルを作ります。

実行したコード

import pandas as pd

df = pd.read_csv("sora2019_8/201908_14_14101010.csv",encoding='shift_jis')

df = df[["測定局コード","日付","時","WD(16Dir)","WS(m/s)","TEMP(℃)"]]

df.to_csv('14101010.csv')

ここまでは成功したのですが

この動作を
201908_14_14101010.csv
201908_14_14101020.csv
201908_14_14102010.csv
・・・

と実行していく繰り返し構文をつくるにはどうすればいいですか。

Answer 1

globで指定フォルダ配下の全ファイルのリストを取得
正規表現でファイル名の指定文字列を取得
出力

こんな感じかと

import pandas as pd
import glob
import re

for f in glob.glob('sora2019_8/*.csv'):
  df = pd.read_csv(f, encoding='shift_jis')
  df = df[["測定局コード","日付","時","WD(16Dir)","WS(m/s)","TEMP(℃)"]]

  m = re.search('_(\d+.csv)', f)
  df.to_csv(m.group(1))

Comments

[o-fk]

実行できました。ありがとうございました！

[o-fk]

すみません、補足で伺いたいのですが、出力先csvのヘッダーの日本語が文字化けしていました。どのように対処すれば良いでしょうか。。。

[aokikenichi]

https://note.nkmk.me/python-pandas-to-csv/
などご覧ください

Pythonは特に設定しなければUTF-8です
お使いのPCがWindowsならばShift-JISかCP932です（どちらもほぼ同じものと考えてとりあえずは大丈夫です）
MacならばMac専用の文字コードが標準と思います

encodingを使って指定するとよろしいかと思います

Answer 2

以下のような感じでできそうです。

for file in ['201908_14_14101010.csv', '201908_14_14101020.csv', '201908_14_14102010.csv']:
    df = pd.read_csv(file, ～)
    ：
    out_file = file.split('_')[2]
    #print(out_file)
    df.to_csv(out_file)

Comments

[o-fk]

ありがとうございます！