有沒有更有效的方式將Pandas DataFrames存儲到CSV？

Question

我編寫了一個小的Python腳本，該腳本對CSV文件執行操作並存儲新修改的腳本。 我想知道是否可以利用任何功能或模塊來提高效率。 這是腳本：

import pandas as pd
import os


print("Current directory is:\n" + os.getcwd() + '\n')
csv = input("Please enter csv file name: ")
csv_list = csv.split('/')

df = pd.read_csv(csv)
df.drop(df[df['is_reply_to'] == 1].index, inplace=True)
df.to_csv('./' + csv_list[-2] + '/' + 'new_' + csv_list[-1])

輸入示例： ./upper_directory/testing.csv

輸出示例： new_testing.csv

我假設目標CSV文件位於當前目錄內的目錄中，因此我使用的方法非常具體。 我想知道是否有某種方法可以使它更通用，因為我不必做諸如csv_list[-2] + '/' + ...類的事情。

謝謝！

Answer 1

您可以這樣創建外觀更好的路徑：

import os
# Directory path of input, then actual file name of path.
out_path = os.path.join(os.path.dirname(csv), 'new_{}'.format(os.path.basename(csv)))
df.to_csv(out_path)