下面是一个示例代码,可以实现按subnumber在CSV文件中的第三行切割每2行,并创建一个百分比列。
import csv
def split_and_add_percentage(filename):
with open(filename, 'r') as file:
reader = csv.reader(file)
rows = list(reader)
# 获取第三行的subnumber
subnumber = rows[2][0]
# 切割每2行
sub_rows = [rows[i:i+2] for i in range(3, len(rows), 2)]
# 计算每个切片的行数
row_counts = [len(sub_row) for sub_row in sub_rows]
# 计算每个切片所占的百分比
percentages = [(count / sum(row_counts)) * 100 for count in row_counts]
# 在每个切片的末尾添加百分比列
for i, sub_row in enumerate(sub_rows):
for row in sub_row:
row.append("{:.2f}%".format(percentages[i]))
# 将修改后的数据写回原始的CSV文件
with open(filename, 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(rows)
# 使用示例:
filename = "data.csv"
split_and_add_percentage(filename)
注意:上述代码假设CSV文件中的第一行是标题行,数据从第二行开始。如果数据从第一行开始,请将range(3, len(rows), 2)
改为range(2, len(rows), 2)
。另外,代码中的百分比保留两位小数。如果想要更多或更少的小数位数,请相应地修改"{:.2f}%"
中的数字。
上一篇:按算术运算符拆分字符串
下一篇:按随机排序