Python 统计 txt 文件第二列不同值并用树状图可视化

假设我们的 txt 文件名为 'data.txt'，内容如下：

apple,red
banana,yellow
apple,green
orange,orange
banana,brown
apple,yellow

我们可以按照以下步骤实现：

读取 txt 文件，将第二列的值存入一个列表中：

data = []
with open('data.txt', 'r') as f:
    for line in f:
        line = line.strip().split(',')
        data.append(line[1])

统计第二列不同值的数量：

count = {}
for d in data:
    if d not in count:
        count[d] = 1
    else:
        count[d] += 1

使用 pyecharts 库绘制树状图：

from pyecharts import options as opts
from pyecharts.charts import Tree

tree_data = []
for key, value in count.items():
    tree_data.append({'name': key, 'value': value})

tree = Tree()
tree.add('', tree_data)
tree.set_global_opts(title_opts=opts.TitleOpts(title='Second Column Count'))
tree.render('second_column_count.html')

最后，我们可以在本地生成一个名为 'second_column_count.html' 的文件，用浏览器打开即可看到可视化结果。