python フォルダを指定してcsvデータを出力

Question

pythonを使用して指定したフォルダにcsvデータを出力したいと考えています。
現在は出力するすべてのcsvデータが全て同じフォルダに出力されているのですが、これをcsvデータの名前に応じて出力先のフォルダを指定したいです。

import pandas as pd
import numpy as np


for list_no in range(1,1001):
    df = pd.read_csv(f"list{list_no}.csv")
    label = pd.cut(df['time'], bins=np.arange(0, 10.5, 0.5), right=False, labels=False)

    limit = 5
    for i in range(0,20):
        drop_size = min((label == i).sum(), limit)
        drop_index = df.loc[label==i,'time'].sort_values().index[:drop_size]
        df = df.drop(drop_index)
        new_file = f"list{list_no}_s2_{(i+1)*0.5:.1f}f.csv"
        df.to_csv(new_file)
        limit -= drop_size    

上記のスクリプトを実行すると、
list1_s2.csv
list1_s2_0.5f.csv
list1_s2_1.0f.csv
.
list1_s2_10.0f.csv
・
・
・
・
list1000_s2.csv
list1000_s2_0.5f.csv
list1000_s2_1.0f.csv
.
list1000_s2_10.0f.csv

といったようなcsvデータが同一のフォルダ内に出力されてしまい、データを操作することに苦戦しております。

list_s2
list_s2_0.5f
list_s2_1.0f
・
・
list_s2_10.0f
といった名前のフォルダを事前に作成しておき、

list_s2のフォルダにはlist1_s2.csv～list1000_s2.csvのデータを、
list_s2_0.5fのフォルダにはlist1_s2_0.5f.csv～list1000_s2_0.5f.csvのデータを
list_s2_10.0fのフォルダにはlist1_s2_10.0f.csv～list1000_s2_10.0f.csvのデータを
といったようにcsvデータを出力する為にはどのように変更したらよろしいでしょうか。

ご教授いただけましたら、幸いです。
よろしくお願いいたします。

Comments

[meg_]

｢ファイルが全て同一ファイル内に出力される｣とはどういう意味でしょうか？

[退会済みユーザー]

出力された新たなcsvデータがすべて同じファイルに出力されるという意味になります。
出力されるcsvデータ名で出力されるファイルを変えたいです。

Answer 1

動作確認してないけど、こんな感じ？

import os
import pandas as pd
import numpy as np


for list_no in range(1,1001):
    out_dir = f"list{list_no}"
    if not os.path.exists(out_dir):
        os.mkdir(out_dir)
    df = pd.read_csv(f"list{list_no}.csv")
    label = pd.cut(df['time'], bins=np.arange(0, 10.5, 0.5), right=False, labels=False)

    limit = 5
    for i in range(0,20):
        drop_size = min((label == i).sum(), limit)
        drop_index = df.loc[label==i,'time'].sort_values().index[:drop_size]
        df = df.drop(drop_index)
        new_file = os.path.join(out_dir, f"list{list_no}_s2_{(i+1)*0.5:.1f}f.csv")
        df.to_csv(new_file)
        limit -= drop_size

Comments

[退会済みユーザー]

ありがとうございます。
解決いたしました！！

[shiracamus]

要求にあってなかったので、削除依頼し、別回答を書きました。

Answer 2

これでどうかな？

import os
import pandas as pd
import numpy as np


for list_no in range(1,1001):
    df = pd.read_csv(f"list{list_no}.csv")
    label = pd.cut(df['time'], bins=np.arange(0, 10.5, 0.5), right=False, labels=False)

    limit = 5
    for i in range(0,20):
        drop_size = min((label == i).sum(), limit)
        drop_index = df.loc[label==i,'time'].sort_values().index[:drop_size]
        df = df.drop(drop_index)
        out_dir = f"list_s2_{(i+1)*0.5:.1f}f"
        if not os.path.exists(out_dir):
            os.mkdir(out_dir)
        out_file = f"list{list_no}_s2_{(i+1)*0.5:.1f}f.csv"
        df.to_csv(os.path.join(out_dir, out_file))
        limit -= drop_size