python - 如何在 Python 中将 CSV 列数据拆分为两列
问题描述
我有以下代码(如下),它抓取 CSV 文件并将数据合并到一个合并的 CSV 文件中。
我现在需要从其中一列中获取特定信息,然后将该信息添加到另一列。
我现在拥有的是一个 output.csv 文件,其中包含以下示例数据:
ID,Name,Flavor,RAM,Disk,VCPUs
45fc754d-6a9b-4bde-b7ad-be91ae60f582,customer1-test1-dns,m1.medium,4096,40,2
83dbc739-e436-4c9f-a561-c5b40a3a6da5,customer2-test2,m1.tiny,128,1,1
ef68fcf3-f624-416d-a59b-bb8f1aa2a769,customer3-test3-dns-api,m1.medium,4096,40,2
我需要做的是打开这个 CSV 文件并将 Name 列中的数据拆分为两列,如下所示:
ID,Name,Flavor,RAM,Disk,VCPUs,Customer,Misc
45fc754d-6a9b-4bde-b7ad-be91ae60f582,customer1-test1-dns,m1.medium,4096,40,2,customer1,test1-dns
83dbc739-e436-4c9f-a561-c5b40a3a6da5,customer2-test2,m1.tiny,128,1,1,customer2,test2
ef68fcf3-f624-416d-a59b-bb8f1aa2a769,customer3-test3-dns-api,m1.medium,4096,40,2,customer3,test3-dns-api
请注意 Misc 列如何可以将多个值拆分为一个或多个-
。
我怎样才能通过 Python 实现这一点。以下是我现在拥有的代码:
import csv
import os
import pandas as pd
by_name = {}
with open('flavor.csv') as b:
for row in csv.DictReader(b):
name = row.pop('Name')
by_name[name] = row
with open('output.csv', 'w') as c:
w = csv.DictWriter(c, ['ID', 'Name', 'Flavor', 'RAM', 'Disk', 'VCPUs'])
w.writeheader()
with open('instance.csv') as a:
for row in csv.DictReader(a):
try:
match = by_name[row['Flavor']]
except KeyError:
continue
row.update(match)
w.writerow(row)
解决方案
试试这个:
import pandas as pd
df = pd.read_csv('flavor.csv')
df[['Customer','Misc']] = df.Name.str.split('-', n=1, expand=True)
df
输出:
ID Name Flavor RAM Disk VCPUs Customer Misc
0 45fc754d-6a9b-4bde-b7ad-be91ae60f582 customer1-test1-dns m1.medium 4096 40 2 customer1 test1-dns
1 83dbc739-e436-4c9f-a561-c5b40a3a6da5 customer2-test2 m1.tiny 128 1 1 customer2 test2
2 ef68fcf3-f624-416d-a59b-bb8f1aa2a769 customer3-test3-dns-api m1.medium 4096 40 2 customer3 test3-dns-api
推荐阅读
- gulp - 使用 gulp 系列时从未定义的任务
- r - R - 用 dplyr 中的特定值替换变量
- c++ - 如何不使用命令行参数编译 CMakeLists.txt 的一部分?
- perforce - Perforce 工作区创建问题
- ubuntu - 如何修复“ModuleNotFoundError:没有名为 'setuptools' 的模块”
- python - 如何在由子进程提供支持的 sudo 命令中输入密码?
- regex - 如何使用正则表达式消除 url 参数的结尾?
- vb.net - 找不到 System.Object VB.NET
- asp.net-core - 如何为字母分页编写标签助手
- java - 计算从旋转字符串中找到原始字符串所需的旋转次数的替代方法