补全有重复值的信息

# coding:utf-8
import csv
ids_input,datas_input,colomn_input = '', '', ''
ids_format,datas_format,column_format = '', '', []
final_data = {}
# 输入包含重复值的key
ids = input('全部id：')
# 将换行输入的数据变成特定格式字符串
for line in iter(input, ids_input):
    ids_format += line + '\n'
# 输入不包含重复值的对应数据
datas = input('完整数据：')
for line in iter(input, datas_input):
    datas_format+= line + '\n'
# 输入完整列名，方便输出
column_num = input('完整列名：')
for line in iter(input, colomn_input):
    column_format.append(line)
# 将输入值格式化为数组
id = ids_format.split()
data = datas_format.split()
# 完整数据变成 key->value 字典
key, value = '', []
for i in range(len(data)):
    column_num = len(column_format)
    print(key)
    if (i+1)%column_num == 1:
        key = data[i]
    elif (i+1)%column_num == 0:
        value.append(data[i])
        print(key)
        final_data[key] = value
        value = []
    else:
        value.append(data[i])
print(final_data)
#python2可以用file替代open
with open("names.csv","w", newline='') as csvfile:
    writer = csv.writer(csvfile)
    #先写入columns_name
    writer.writerow(column_format)
    list = []
    #循环所有id 补全信息并输出
    for i, v in enumerate(id):
        list.append(v)
        f_list = list+final_data[v]
        writer.writerow(f_list)
        list = []
补全有重复值的信息
csv文件内容

excel打开