如何从 python 中的 CSV 列中捕获特定信息

Question

我有一个 CSV 文件，其中 table 包含我想在另一个 table 中引用的信息。为了给你一个更好的视角，我有以下例子：

"ID","Name","Flavor"
"45fc754d-6a9b-4bde-b7ad-be91ae60f582","account1-test1","m1.medium"
"83dbc739-e436-4c9f-a561-c5b40a3a6da5","account3-test2","m1.tiny"
"ef68fcf3-f624-416d-a59b-bb8f1aa2a769","account1-test3","m1.medium"

我想添加另一列引用名称列并将客户名称拉入一列并将其余信息拉入另一列，示例：

"ID","Name","Flavor","Customer","Misc"
"45fc754d-6a9b-4bde-b7ad-be91ae60f582","account1-test1","m1.medium","account1","test1"
"83dbc739-e436-4c9f-a561-c5b40a3a6da5","account3-test2","m1.tiny","account3,"test2"
"ef68fcf3-f624-416d-a59b-bb8f1aa2a769","account1-test3","m1.medium","account1","test3"

此处的任务是使用 python 脚本打开原始 CSV 文件，并使用添加的列创建新的 CSV 文件。有任何想法吗？我一直无法成功解析名称列。

Answer 1

您是否尝试过使用 pandas DataFrame 打开您的 csv 文件？这可以通过以下方式完成：

df = pd.read_csv('input_data.csv')

如果 customer 和 misc 列是另一个 csv 文件的一部分，您可以使用与上述相同的方法加载它（命名 df2），然后附加以下内容：

df['Customer'] = df2['Customer']

然后您可以使用以下内容将 DataFrame 输出为 csv 文件：

df.to_csv('output_data_name.csv')

Answer 2

data = pd.read_csv('your_file.csv')
data[['Customer','Misc']] = data.Name.str.split("-",expand=True)

现在您可以再次将其保存到 csv 文件中：

data.to_csv('another_file.csv')

如何从 python 中的 CSV 列中捕获特定信息

How to capture specific info from CSV column in python

python

csv

export-to-csv