Pandas系列（十四）- 实战案例

import pandas as pd
import string

#创建Series的两种方式
#方式一
t = pd.Series([1,2,3,4,43],index=list(‘asdfg‘))
print(t)
#方式二
temp_dict = {‘name‘:‘xiaohong‘,‘age‘:30,‘tel‘:10086}
t2 = pd.Series(temp_dict)
print(t2)
#字典推导式
a = {string.ascii_uppercase[i]:i for i in range(10)}
print(a)
print(pd.Series(a))
print(pd.Series(a,index=list(string.ascii_uppercase[5:15])))　

import pandas as pd
from pymongo import MongoClient

#pandas读取csv文件
# df = pd.read_csv(‘dogNames2.csv‘)
# print(df)

client = MongoClient()
collection = client[‘meipai‘][‘meipai_video‘]
data = collection.find()
data_list = []
for i in data:
    temp = {}
    temp[‘cut_url‘] = i[‘cut_url‘]
    temp[‘create_time‘] = i[‘create_time‘]
    temp[‘title‘] = i[‘title‘]
    temp[‘video_url‘] = i[‘video_url‘]
    data_list.append(temp)

# print(data)
# t1 = data[0]
# t1 = pd.Series(t1)
# print(t1)

df = pd.DataFrame(data_list)
print(df.info())
print(df.describe())
# print(df.head())
# print(‘*‘*100)
# print(df.tail())

import pandas as pd

temp_dict = {‘name‘:[‘xiaohong‘,‘xiaozhang‘],‘age‘:[30,23],‘tel‘:[10086,10010]}
t1 = pd.DataFrame(temp_dict)
print(t1)

temp_dict1 = [{‘name‘:‘xiaohong‘,‘age‘:23,‘tel‘:10086},{‘name‘:‘xiaogang‘,‘age‘:12},{‘name‘:‘xiaozhang‘,‘tel‘:10010}]
t2 = pd.DataFrame(temp_dict1)
print(t2)

import pandas as pd

#pandas读取csv文件
df = pd.read_csv(‘dogNames2.csv‘)
# print(df.head())
# print(df.info())

#DataFrame中排序的方法
df = df.sort_values(by=‘Count_AnimalName‘,ascending=False)
# print(df.head())

#pandas取行和列的注意事项
# - 方括号写数组，表示取行，对行进行操作
# - 写字符串，表示取列索引，对列进行操作
print(df[:20])
print(df[:20][‘Row_Labels‘])
print(type(df[‘Row_Labels‘]))

#bool索引
print(df[(df[‘Row_Labels‘].str.len()>4)&(df[‘Count_AnimalName‘]>800)])

import pandas as pd
from matplotlib import pyplot as plt

file_path = ‘./IMDB-Movie-data.csv‘
df = pd.read_csv(file_path)
# print(df.head(1))
# print(df.info())

# rating,runtime分布情况
# 选择图形：直方图
# 准备数据
runtime_data = df[‘Runtime (Minutes)‘].values
max_runtime = runtime_data.max()
min_runtime = runtime_data.min()

#计算组距
num_bin = (max_runtime-min_runtime)//5

#设置图行大小
plt.figure(figsize=(13,6),dpi=80)
#画直方图
plt.hist(runtime_data,num_bin)

plt.xticks(range(min_runtime,max_runtime+5,5))

#显示
plt.show()

import pandas as pd
from matplotlib import pyplot as plt
from functools import reduce

file_path = ‘./IMDB-Movie-data.csv‘
df = pd.read_csv(file_path)
# print(df.head(1))
# print(df.info())

# rating,runtime分布情况
# 选择图形：直方图
# 准备数据
# runtime_data = df[‘Runtime (Minutes)‘].values
rate_data = df[‘Rating‘].values
max_rate = rate_data.max()
min_rate = rate_data.min()

#设置不等宽组距，hist方法中取到的会是一个左闭右开的区间[1,9,3.5)
num_bin_list = [1.9,3.5]
i = 3.5
while i<=max_rate:
    i += 0.5
    num_bin_list.append(i)
print(num_bin_list)

#设置图形大小
plt.figure(figsize=(13,6),dpi=80)
#画直方图
plt.hist(rate_data,num_bin_list)

#xticks让之前的组距能够对上
plt.xticks(num_bin_list)

#显示
plt.show()

[1.9, 3.5, 4.0, 4.5, 5.0, 5.5, 6.0, 6.5, 7.0, 7.5, 8.0, 8.5, 9.0, 9.5]

import numpy
import pandas as pd
df = pd.read_csv(‘IMDB-Movie-Data.csv‘)
print(df.info())
print(df.describe())
#获取评分的均分
rate_mean = df.Rating.mean()
print(rate_mean)
#获取导演的人数
print(df.Director.value_counts().count())
print(len(set(df.Director.tolist())))
print(len(df.Director.unique()))
#获取演员的人数
temp_actors_list = df.Actors.str.split(‘,‘).tolist()
actors_list = [i for j in temp_actors_list for i in j]
# numpy.array(temp_actors_list).flatten()
actors_num = len(set(actors_list))
print(actors_num)
<class ‘pandas.core.frame.DataFrame‘>
RangeIndex: 1000 entries, 0 to 999
Data columns (total 12 columns):
Rank                  1000 non-null int64
Title                 1000 non-null object
Genre                 1000 non-null object
Description           1000 non-null object
Director              1000 non-null object
Actors                1000 non-null object
Year                  1000 non-null int64
Runtime (Minutes)     1000 non-null int64
Rating                1000 non-null float64
Votes                 1000 non-null int64
Revenue (Millions)    872 non-null float64
Metascore             936 non-null float64
dtypes: float64(3), int64(4), object(5)
memory usage: 93.8+ KB
None
              Rank         Year     ...      Revenue (Millions)   Metascore
count  1000.000000  1000.000000     ...              872.000000  936.000000
mean    500.500000  2012.783000     ...               82.956376   58.985043
std     288.819436     3.205962     ...              103.253540   17.194757
min       1.000000  2006.000000     ...                0.000000   11.000000
25%     250.750000  2010.000000     ...               13.270000   47.000000
50%     500.500000  2014.000000     ...               47.985000   59.500000
75%     750.250000  2016.000000     ...              113.715000   72.000000
max    1000.000000  2016.000000     ...              936.630000  100.000000
[8 rows x 7 columns]

# -*- coding: utf-8 -*-

"""
@Datetime: 2018/11/19
@Author: Zhang Yafei
"""
"""
对于这一组电影数据，如果我们希望统计电影分类(genre)的情况，应该如何处理数据？
思路：重新构造一个全为0的数组，列名为分类，如果某一条数据中分类出现过，就让0变为1
"""
import numpy as np
import pandas as pd
from matplotlib import pyplot as plt
from matplotlib import font_manager

#中文字体
my_font = font_manager.FontProperties(family=‘SimHei‘)
#显示完整的列
pd.set_option(‘display.max_columns‘, None)

df = pd.read_csv(‘IMDB-Movie-Data.csv‘)
#统计分类列表
temp_list = df.Genre.str.split(‘,‘).tolist()
genre_list = list(set([i for j in temp_list for i in j]))

#构造全为0的数组
zero_df = pd.DataFrame(np.zeros((df.shape[0],len(genre_list))),columns=genre_list)
# print(zero_df)
#给每个电影出现分类的位置赋值1
for i in range(df.shape[0]):
    zero_df.loc[i,temp_list[i]] = 1

# print(zero_df.head(1))
genre_count = zero_df.sum(axis=0)
print(genre_count)

#排序
genre_count = genre_count.sort_values()
_x = genre_count.index
_y = genre_count.values
#画图
plt.figure(figsize=(15,6),dpi=80)
plt.bar(range(len(_x)),_y,width=0.4,color="orange")
plt.xticks(range(len(_x)),_x)
plt.title(‘电影分类统计图‘,fontproperties=my_font)
plt.show()

# -*- coding: utf-8 -*-

"""
@Datetime: 2018/11/19
@Author: Zhang Yafei
"""
"""
现在我们有一组关于全球星巴克店铺的统计数据，如果我想知道美国的星巴克数量和中国的哪个多，或者我想知道中国每个省份星巴克的数量的情况，那么应该怎么办？
思路：遍历一遍，每次加1 ？？？
"""
import pandas as pd

pd.set_option(‘display.max_columns‘, None)

df = pd.read_csv(‘starbucks_store_worldwide.csv‘)
# print(df.head(1))
# print(df.info())
grouped = df.groupby(by=‘Country‘)
# print(grouped)

# DataFrameGroupBy
# 可以进行遍历
# for i,j in grouped:
#     print(i)
#     print(‘-‘*100)
#     print(j)
#     print(‘*‘*100)
country_count = grouped[‘Brand‘].count()
# print(country_count[‘US‘])
# print(country_count[‘CN‘])

#统计中国每个省份店铺的数量
china_data = df[df.Country == ‘CN‘]
china_grouped = china_data.groupby(by=‘State/Province‘).count()[‘Brand‘]
# print(china_grouped)
#数据按照多个条件进行分组
brand_grouped = df[‘Brand‘].groupby(by=[df[‘Country‘],df[‘State/Province‘]]).count()
# print(brand_grouped)
# print(type(brand_grouped))
#数据按照多个条件进行分组,返回dataframe
brand_grouped1 = df[[‘Brand‘]].groupby(by=[df[‘Country‘],df[‘State/Province‘]]).count()
brand_grouped2 = df.groupby(by=[df[‘Country‘],df[‘State/Province‘]])[[‘Brand‘]].count()
brand_grouped3 = df.groupby(by=[df[‘Country‘],df[‘State/Province‘]]).count()[[‘Brand‘]]
# print(brand_grouped1)
# print(brand_grouped2)
# print(brand_grouped3)
#索引的方法和属性
print(brand_grouped1)
print(brand_grouped1.index)

import pandas as pd
from matplotlib import pyplot as plt

pd.set_option(‘display.max_columns‘, None)

df = pd.read_csv(‘starbucks_store_worldwide.csv‘)

df = df.groupby(by=‘Country‘).count()[‘Brand‘].sort_values(ascending=False)[:10]

_x = df.index
_y = df.values

#画图
plt.figure(figsize=(13,6),dpi=80)

plt.bar(_x,_y)

plt.show()

import pandas as pd
from matplotlib import pyplot as plt
from matplotlib import font_manager

my_font = font_manager.FontProperties(family=‘SimHei‘)

pd.set_option(‘display.max_columns‘, None)

df = pd.read_csv(‘starbucks_store_worldwide.csv‘)
df = df[df[‘Country‘]==‘CN‘]
print(df.head(1))

df = df.groupby(by=‘City‘).count()[‘Brand‘].sort_values(ascending=False)[:25]

_x = df.index
_y = df.values

#画图
plt.figure(figsize=(13,6),dpi=80)

# plt.bar(_x,_y,width=0.3,color=‘orange‘)
plt.barh(_x,_y,height=0.3,color=‘orange‘)

# plt.xticks(_x,fontproperties=my_font)
plt.yticks(_x,fontproperties=my_font)

plt.show()

import pandas as pd
from matplotlib import pyplot as plt

pd.set_option(‘display.max_columns‘, None)

df = pd.read_csv(‘books.csv‘)
# print(df.info())
data = df[pd.notnull(df[‘original_publication_year‘])]
grouped = data.groupby(by=‘original_publication_year‘).count()[‘title‘]
# print(grouped)

grouped1 = data.average_rating.groupby(by=data[‘original_publication_year‘]).mean()
# print(grouped1)

_x = grouped1.index
_y = grouped1.values

plt.figure(figsize=(15,6),dpi=80)
plt.plot(range(len(_x)),_y)
plt.xticks(range(len(_x))[::10],_x[::10].astype(int),rotation=45)
plt.show()

import pandas as pd
from matplotlib import pyplot as plt
import numpy as np

pd.set_option(‘display.max_columns‘,None)

df = pd.read_csv(‘911.csv‘)
# print(df.head(1))
# print(df.info())

#获取分类
temp_list = df.title.str.split(‘:‘).tolist()
cate_list = list(set([i[0] for i in temp_list]))
# print(cate_list)

#构造全为0的数组
zeros_df = pd.DataFrame(np.zeros((df.shape[0],len(cate_list))),columns=cate_list)

#赋值
for cate in cate_list:
    zeros_df[cate][df.title.str.contains(cate)] = 1
print(zeros_df)

sum_ret = zeros_df.sum(axis=0)
print(sum_ret)　　

import pandas as pd
from matplotlib import pyplot as plt
import numpy as np

pd.set_option(‘display.max_columns‘,None)

df = pd.read_csv(‘911.csv‘)
# print(df.head(1))
# print(df.info())

#获取分类
temp_list = df.title.str.split(‘:‘).tolist()
cate_list = [i[0] for i in temp_list]

df[‘cate‘] = pd.DataFrame(np.array(cate_list).reshape(df.shape[0],1))
print(df.groupby(by=‘cate‘).count()[‘title‘])　

# -*- coding: utf-8 -*-

"""
@Datetime: 2018/11/19
@Author: Zhang Yafei
"""
"""
统计出911数据中不同月份电话次数的变化情况
"""
import pandas as pd
from matplotlib import pyplot as plt
import numpy as np


pd.set_option(‘display.max_columns‘,None)

df = pd.read_csv(‘911.csv‘)
df.drop_duplicates()
df.timeStamp = pd.to_datetime(df.timeStamp)  #时间字符串转时间格式

df.set_index(‘timeStamp‘,inplace=True)  #设置时间格式为索引
# print(df.head())

#统计出911数据中不同月份电话次数
count_by_month = df.resample(‘M‘).count()[‘title‘]
print(count_by_month)

#画图
_x = count_by_month.index
_y = count_by_month.values

plt.figure(figsize=(15,8),dpi=80)

plt.plot(range(len(_x)),_y)

plt.xticks(range(len(_x)),_x.strftime(‘%Y-%m-%d‘),rotation=45)

plt.show()

# -*- coding: utf-8 -*-

"""
@Datetime: 2018/11/19
@Author: Zhang Yafei
"""
"""
统计出911数据中不同月份不同类型的电话的次数的变化情况
"""
import pandas as pd
from matplotlib import pyplot as plt
import numpy as np


pd.set_option(‘display.max_columns‘,None)

df = pd.read_csv(‘911.csv‘)
#把时间字符串转化为时间类型设置为索引
df.timeStamp = pd.to_datetime(df.timeStamp)

#添加列，表示分类
temp_list = df.title.str.split(‘:‘).tolist()
cate_list = [i[0] for i in temp_list]
df[‘cate‘] = pd.DataFrame(np.array(cate_list).reshape(df.shape[0],1))

df.set_index(‘timeStamp‘,inplace=True)

plt.figure(figsize=(15, 8), dpi=80)

#分组
for group_name,group_data in df.groupby(by=‘cate‘):
    #对不同的分类都进行绘图
    count_by_month = group_data.resample(‘M‘).count()[‘title‘]
    # 画图
    _x = count_by_month.index
    _y = count_by_month.values
    plt.plot(range(len(_x)),_y,label=group_name)

plt.xticks(range(len(_x)), _x.strftime(‘%Y-%m-%d‘), rotation=45)

plt.legend(loc=‘best‘)
plt.show()

# -*- coding: utf-8 -*-

"""
@Datetime: 2018/11/19
@Author: Zhang Yafei
"""
"""
绘制美国和中国PM2.5随时间的变化情况
"""
import pandas as pd
from matplotlib import pyplot as plt

pd.set_option(‘display.max_columns‘,None)

df = pd.read_csv(‘PM2.5/BeijingPM20100101_20151231.csv‘)
# print(df.head())

#把分开的时间字符串通过periodIndex的方法转化为pandas的时间类型
period = pd.PeriodIndex(year=df.year,month=df.month,day=df.day,hour=df.hour,freq=‘H‘)
df[‘datetime‘] = period
print(df.head(10))

#把datetime设置为索引
df.set_index(‘datetime‘,inplace=True)

#进行降采样
df = df.resample(‘7D‘).mean()

#处理缺失值，删除缺失数据
# data = df[‘PM_US Post‘].dropna()
# china_data = df[‘PM_Nongzhanguan‘].dropna()
data = df[‘PM_US Post‘]
china_data = df[‘PM_Nongzhanguan‘]

#画图
_x = data.index
_y = data.values

_x_china = china_data.index
_y_china = china_data.values

plt.figure(figsize=(13,8),dpi=80)

plt.plot(range(len(_x)),_y,label=‘US_POST‘,alpha=0.7)
plt.plot(range(len(_x_china)),_y_china,label=‘CN_POST‘,alpha=0.7)

plt.xticks(range(0,len(_x_china),10),list(_x_china.strftime(‘%Y%m%d‘))[::10],rotation=45)

plt.show()