登录
首页 >  文章 >  python教程

如何从包含多列数据的 CSV 文件中合并特定列并计算各个选项出现的比例?

时间:2024-11-01 21:57:47 344浏览 收藏

珍惜时间,勤奋学习!今天给大家带来《如何从包含多列数据的 CSV 文件中合并特定列并计算各个选项出现的比例?》,正文内容主要涉及到等等,如果你正在学习文章,或者是对文章有疑问,欢迎大家关注我!后面我会持续更新相关内容的,希望都能帮到正在学习的大家!

如何从包含多列数据的 CSV 文件中合并特定列并计算各个选项出现的比例?

csv 文件合并与内容比例计算

问题:

如何从包含多列数据的 csv 文件中读取内容并合并特定的列,同时计算各个选项出现的比例?

解决方案:

使用 python 的 pandas 库来处理 csv 文件:

import pandas as pd
from collections import Counter

# 读取 CSV 文件
df = pd.read_csv('test.csv', encoding='cp932')

# 获取 CMD 列作为主键
index_list_cmd = df['cmd'].tolist()

# 计算 CMD 出现的次数
idxLstCmd_cnt = Counter(list(index_list_cmd))

# 去除重复的 CMD
index_list_cmd = sorted(set(index_list_cmd), key=list(index_list_cmd).index)

# 计算每个 CMD 出现的次数
cntLst = []
for lst_i in range(len(index_list_cmd)):
    cntLst.append(idxLstCmd_cnt[index_list_cmd[lst_i]])

# 合并 OPT 列
index_list_opt_idx = []

# 删除之前保存的文件
if os.path.isfile(useRateF):
    os.remove(useRateF)

# 遍历 CMD
for cmd_i in range(len(index_list_cmd)):
    # 获得 OPT 列的 index
    index_list_opt_idx = list(df['opt'][df['cmd'] == index_list_cmd[cmd_i]].index.values)

    index_list_opt = []

    # 遍历 OPT
    for opt_i in range(len(index_list_opt_idx)):
        try:
            index_list_opt.append(df.loc[int(index_list_opt_idx[opt_i]), 'opt'])
        except:
            print(index_list_cmd[cmd_i])
            print(index_list_opt_idx[opt_i])
            print(opt_i)

    # 列表转字符串
    str_idxLstOpt = ' '.join(index_list_opt)

    # 字符串转回列表
    lst_idxLstOpt = str_idxLstOpt.split(' ')

    # 计算 OPT 出现的比例
    for tag in sorted(set(lst_idxLstOpt), key=lst_idxLstOpt.index):
        with open(useRateF, 'a', encoding='utf-8_sig') as uf:
            uf.writelines(str(index_list_cmd[cmd_i]) + ',' + tag + ',' + str(lst_idxLstOpt.count(tag)/cntLst[cmd_i]))
            uf.writelines('\n')

这个脚本将读取 csv 文件,合并 opt 列,并输出每个 cmd 选项出现的比例。

今天关于《如何从包含多列数据的 CSV 文件中合并特定列并计算各个选项出现的比例?》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>