'''
问题:
1、假设DataFrame中有一列名为type,其字段中内容为a,b,c 等用,隔开的值,如:
type
a,b,c
a,f,x
b,c,e
...
统计type中每个类型出现的次数 并绘图
'''
import pandas as pd
import numpy as np
from matplotlib import pyplot as plt
# 读取数据
df = pd.read_csv("文件路径/文件名")
#统计分类的列表
temp_list = df["type"].str.split(",").tolist()
type_list = list(set([i for j in temp_list for i in j]))
# 构造全为0的数组
zeros_df = pd.DataFrame(np.zeros((df.shape[0],len(type_list))),columns=type_list)
# print(zeros_df)
# 给每个type出现分类的位置赋值为1
for i in range(df.shape[0]):
zeros_df.loc[i,temp_list[i]] = 1
# 统计每个分类type的数量和
type_count = zeros_df.sum(axis=0)
print(type_count)
# 排序
type_count = type_count.sor_values()
_x = type_count.index
_y = type_count.values
# 绘图
# 设置图片大小
plt.figure(figsize=(20,8),dpi=80)
plt.bar(range(len(_x)),_y) #绘制条形图
plt.xticks(range(_x),_x)
plt.show()
复制代码
series练习
最新推荐文章于 2024-03-14 23:16:34 发布