数据科学库--第六天
应用
统计911报警电话原因的分类
第一种方式:构建全为0的Dataframe
import pandas as pd
import numpy as np
from matplotlib import pyplot as plt
file_path = "./911.csv"
df = pd.read_csv(file_path)
#获取分类
temp_list = df["title"].str.split(":").tolist()
cate_list = list(set([i[0] for i in temp_list]))
print(cate_list)
#g构造全为0的数据
zeros_df = pd.DataFrame(np.zeros((df.shape[0],len(cate_list))),columns=cate_list)
#给全0数组赋值,对zeros_df的三列进行遍历处理,相较于对整个temp_list的25万行进行遍历要快的多
for cate in cate_list:
zeros_df[cate][df["title"].str.contains(cate)] = 1