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)

有没有更有效的方式将Pandas DataFrames存储到CSV？

1 个答案: