来源:早起Python
本文约5800字,建议阅读15分钟。
本文为你介绍Pandas基础、Pandas数据处理、金融数据处理等方面的一些习题。
Pandas 是基于 NumPy 的一种数据处理工具,该工具为了解决数据分析任务而创建。Pandas 纳入了大量库和一些标准的数据模型,提供了高效地操作大型数据集所需的函数和方法。
和鲸社区的刘早起创作了这个项目,其中包含Pandas基础、Pandas数据处理、金融数据处理、当Pandas遇上NumPy、补充内容 5个部分。在深度和广度上,都相较之前的Pandas习题系列有了很大的提升。
此前的Pandas系列文章:
- 10套练习,教你如何用Pandas做数据分析
- 50道练习带你玩转Pandas
- 给Excel重度用户准备的Pandas教程:用Pandas逐帧还原20个Excel常用操作
以上3套,再加上现在这篇,各种姿势,各种体位基本都已经集齐了
希望榨干我一个,满足大家所有人,靴靴
Part 1 Pandas基础
1.将下面的字典创建为DataFrame
data?=?{"grammer":["Python","C","Java","GO",np.nan,"SQL","PHP","Python"],
???????"score":[1,2,np.nan,4,5,6,7,10]}
???????
df?=?pd.DataFrame(data)
df2.提取含有字符串"Python"的行
#方法一
df[df['grammer']?==?'Python']
#方法二
results?=?df['grammer'].str.contains("Python")
results.fillna(value=False,inplace?=?True)
df[results]3.输出df的所有列名
print(df.columns)4.修改第二列列名为'popularity'
df.rename(columns={'score':'popularity'},?inplace?=?True)
df5.统计grammer列中每种编程语言出现的次数
df['grammer'].value_counts()6.将空值用上下值的平均值填充
df['popularity']?=?df['popularity'].fillna(df['popularity'].interpolate())
df7.提取popularity列中值大于3的行
df[df['popularity']?>?3]8.按照grammer列进行去除重复值
df.drop_duplicates(['grammer'])9.计算popularity列平均值
df['popularity'].mean()10.将grammer列转换为list
df['grammer'].to_list()11.将DataFrame保存为EXCEL
df.to_excel('test.xlsx')12.查看数据行列数
df.shape13.提取popularity列值大于3小于7的行
df[(df['popularity']?>?3)?&?(df['popularity']?7)]14.交换两列位置
#?方法1
temp?=?df['popularity']
df.drop(labels=['popularity'],?axis=1,inplace?=?True)
df.insert(0,?'popularity',?temp)
df
#?方法2
cols?=?df.columns[[1,0]]
df?=?df[cols]
df15.提取popularity列最大值所在行
df[df['popularity']?==?df['popularity'].max()]16.查看最后5行数据
df.tail()17.删除最后一行数据
df.drop([len(df)-1],inplace=True)
df18.添加一行数据['Perl',6.6]
row={'grammer':'Perl','popularity':6.6}
df?=?df.append(row,ignore_index=True)
df19.对数据按照"popularity"列值的大小进行排序
df.sort_values("popularity",inplace=True)
df20.统计grammer列每个字符串的长度
df['grammer']?=?df['grammer'].fillna('R')
df['len_str']?=?df['grammer'].map(lambda?x:?len(x))
dfPart 2 Pandas数据处理
21.读取本地EXCEL数据
import?pandas?as?pd
df?=?pd.read_excel('pandas120.xlsx')22.查看df数据前5行
df.head()23.将salary列数据转换为最大值与最小值的平均值
#备注,在某些版本pandas中.ix方法可能失效,可使用.iloc,参考https://mp.weixin.qq.com/s/5xJ-VLaHCV9qX2AMNOLRtw
#为什么不能直接使用max,min函数,因为我们的数据中是20k-35k这种字符串,所以需要先用正则表达式提取数字
import?re
#?方法一:apply +?自定义函数
def?func(df):
????lst?=?df['salary'].split('-')
????smin?=?int(lst[0].strip('k'))
????smax?=?int(lst[1].strip('k'))
????df['salary']?=?int((smin?+?smax)?/?2?*?1000)
????return?df
df?=?df.apply(func,axis=1)
#?方法二:iterrows +?正则
import?re
for?index,row?in?df.iterrows():
????nums?=?re.findall('\d+',row[2])
????df.iloc[index,2]?=?int(eval(f'({nums[0]}?+?{nums[1]})?/?2?*?1000'))24.将数据根据学历进行分组并计算平均薪资
print(df.groupby('education').mean())25.将createTime列时间转换为月-日
#备注,在某些版本pandas中.ix方法可能失效,可使用.iloc,参考https://mp.weixin.qq.com/s/5xJ-VLaHCV9qX2AMNOLRtw
for?i?in?range(len(df)):
????df.ix[i,0]?=?df.ix[i,0].to_pydatetime().strftime("%m-%d")??
df.head()26.查看索引、数据类型和内存信息
df.info()27.查看数值型列的汇总统计
df.describe()28.新增一列根据salary将数据分为三组
bins?=?[0,5000,?20000,?50000]
group_names?=?['低',?'中',?'高']
df['categories']?=?pd.cut(df['salary'],?bins,?labels=group_names)
df29.按照salary列对数据降序排列
df.sort_values('salary',?ascending=False)30.取出第33行数据
df.loc[32]31.计算salary列的中位数
np.median(df['salary'])32.绘制薪资水平频率分布直方图
#执行两次
df.salary.plot(kind='hist')33.绘制薪资水平密度曲线
df.salary.plot(kind='kde',xlim=(0,80000))34.删除最后一列categories
del?df['categories']
#?等价于
df.drop(columns=['categories'],?inplace=True)35.将df的第一列与第二列合并为新的一列
df['test']?=?df['education']+df['createTime']
df36.将education列与salary列合并为新的一列
#备注:salary为int类型,操作与35题有所不同
df["test1"]?=?df["salary"].map(str)?+?df['education']
df37.计算salary最大值与最小值之差
df[['salary']].apply(lambda?x:?x.max()?-?x.min())38.将第一行与最后一行拼接,成一个新表
pd.concat([df[:1],?df[-2:-1]])39.将第8行数据添加至末尾
df.append(df.iloc[7])40.查看每列的数据类型
df.dtypes41.将createTime列设置为索引
df.set_index("createTime")42.生成一个和df长度相同的随机数dataframe
df1?=?pd.DataFrame(pd.Series(np.random.randint(1,?10,?135)))
df143.将上一题生成的dataframe与df合并
df=?pd.concat([df,df1],axis=1)
df44.生成新的一列new为salary列减去之前生成随机数列
df["new"]?=?df["salary"]?-?df[0]
df45.检查数据中是否含有任何缺失值
df.isnull().values.any()46.将salary列类型转换为浮点数
df['salary'].astype(np.float64)47.计算salary大于10000的次数
len(df[df['salary']>10000])48.查看每种学历出现的次数
df.education.value_counts()49.查看education列共有几种学历
df['education'].nunique()50.提取salary与new列的和大于60000的最后3行
df1?=?df[['salary','new']]
rowsums?=?df1.apply(np.sum,?axis=1)
res?=?df.iloc[np.where(rowsums?>?60000)[0][-3:],?:]
resPart 3 金融数据处理
51.使用绝对路径读取本地Excel数据
#请将下面的路径替换为你存储数据的路径
data?=?pd.read_excel('/Users/Desktop/600000.SH.xls')52.查看数据前三行
data.head(3)53.查看每列数据缺失值情况
data.isnull().sum()54.提取日期列含有空值的行
data[data['日期'].isnull()]55.输出每列缺失值具体行数
for?columname?in?data.columns:
????if?data[columname].count()?!=?len(data):
????????loc?=?data[columname][data[columname].isnull().values==True].index.tolist()
????????print('列名:"{}", 第{}行位置有缺失值'.format(columname,loc))56.删除所有存在缺失值的行
#?备注
# axis:0-行操作(默认),1-列操作
# how:any-只要有空值就删除(默认),all-全部为空值才删除
# inplace:False-返回新的数据集(默认),True-在原数据集上操作
data.dropna(axis=0,?how='any',?inplace=True)57.绘制收盘价的折线图
import?matplotlib.pyplot?as?plt?
plt.style.use('seaborn-darkgrid')?#?设置画图的风格
plt.rc('font',??size=6)?#设置图中字体和大小
plt.rc('figure',?figsize=(4,3),?dpi=150)?#?设置图的大小
data['收盘价(元)'].plot()
#?等价于
import?matplotlib.pyplot?as?plt
plt.plot(df['收盘价(元)'])58.同时绘制开盘价与收盘价
data[['收盘价(元)','开盘价(元)']].plot()59.绘制涨跌幅的直方图
plt.hist(df['涨跌幅(%)'])
#?等价于
df['涨跌幅(%)'].hist()60.让直方图更细致
data['涨跌幅(%)'].hist(bins?=?30)61.以data的列名创建一个dataframe
temp?=?pd.DataFrame(columns?=?data.columns.to_list())62.打印所有换手率不是数字的行
for?i?in?range(len(data)):
????if?type(data.iloc[i,13])?!=?float:
????????temp?=?temp.append(data.loc[i])
temp63.打印所有换手率为--的行
data[data['换手率(%)'].isin(['--'])]64.重置data的行号
data?=?data.reset_index()65.删除所有换手率为非数字的行
k?=[]
for?i?in?range(len(data)):
????if?type(data.iloc[i,13])?!=?float:
????????k.append(i)
data.drop(labels=k,inplace=True)66.绘制换手率的密度曲线
data['换手率(%)'].plot(kind='kde')67.计算前一天与后一天收盘价的差值
data['收盘价(元)'].diff()68.计算前一天与后一天收盘价变化率
data['收盘价(元)'].pct_change()69.设置日期为索引
data?=?data.set_index('日期')70.以5个数据作为一个数据滑动窗口,在这个5个数据上取均值(收盘价)
data['收盘价(元)'].rolling(5).mean()71.以5个数据作为一个数据滑动窗口,计算这五个数据总和(收盘价)
data['收盘价(元)'].rolling(5).sum()72.将收盘价5日均线、20日均线与原始数据绘制在同一个图上
data['收盘价(元)'].plot()
data['收盘价(元)'].rolling(5).mean().plot()
data['收盘价(元)'].rolling(20).mean().plot()73.按周为采样规则,取一周收盘价最大值
data['收盘价(元)'].resample('W').max()74.绘制重采样数据与原始数据
data['收盘价(元)'].plot()
data['收盘价(元)'].resample('7D').max().plot()75.将数据往后移动5天
data.shift(5)76.将数据向前移动5天
data.shift(-5)77.使用expending函数计算开盘价的移动窗口均值
data['开盘价(元)'].expanding(min_periods=1).mean()78.绘制上一题的移动均值与原始数据折线图
data['expanding?Open?mean']=data['开盘价(元)'].expanding(min_periods=1).mean()
data[['开盘价(元)',?'expanding?Open?mean']].plot(figsize=(16,?6))79.计算布林指标
data['former?30?days?rolling?Close?mean']=data['收盘价(元)'].rolling(20).mean()
data['upper?bound']=data['former?30?days?rolling?Close?mean']+2*data['收盘价(元)'].rolling(20).std()#在这里我们取20天内的标准差
data['lower?bound']=data['former?30?days?rolling?Close?mean']-2*data['收盘价(元)'].rolling(20).std()80.计算布林线并绘制
data[['收盘价(元)',?'former?30?days?rolling?Close?mean','upper?bound','lower?bound'?]].plot(figsize=(16,?6))Part 4 当Pandas遇上NumPy
81.导入并查看pandas与numpy版本
import?pandas?as?pd
import?numpy?as?np
print(np.__version__)
print(pd.__version__)82.从NumPy数组创建DataFrame
#备注?使用numpy生成20个0-100随机数
tem?=?np.random.randint(1,100,20)
df1?=?pd.DataFrame(tem)
df183.从NumPy数组创建DataFrame
#备注?使用numpy生成20个0-100固定步长的数
tem?=?np.arange(0,100,5)
df2?=?pd.DataFrame(tem)
df284.从NumPy数组创建DataFrame
#备注?使用numpy生成20个指定分布(如标准正态分布)的数
tem?=?np.random.normal(0,?1,?20)
df3?=?pd.DataFrame(tem)
df385.将df1,df2,df3按照行合并为新DataFrame
df?=?pd.concat([df1,df2,df3],axis=0,ignore_index=True)
df86.将df1,df2,df3按照列合并为新DataFrame
df?=?pd.concat([df1,df2,df3],axis=1,ignore_index=True)
df87.查看df所有数据的最小值、25%分位数、中位数、75%分位数、最大值
print(np.percentile(df,?q=[0,?25,?50,?75,?100]))88.修改列名为col1,col2,col3
df.columns?=?['col1','col2','col3']89.提取第一列中不在第二列出现的数字
df['col1'][~df['col1'].isin(df['col2'])]90.提取第一列和第二列出现频率最高的三个数字
temp?=?df['col1'].append(df['col2'])
temp.value_counts().index[:3]91.提取第一列中可以整除5的数字位置
np.argwhere(df['col1']?%?5==0)92.计算第一列数字前一个与后一个的差值
df['col1'].diff().tolist()93.将col1,col2,clo3三列顺序颠倒
df.ix[:,?::-1]94.提取第一列位置在1,10,15的数字
df['col1'].take([1,10,15])
#?等价于
df.iloc[[1,10,15],0]95.查找第一列的局部最大值位置
#备注?即比它前一个与后一个数字的都大的数字
tem?=?np.diff(np.sign(np.diff(df['col1'])))
np.where(tem?==?-2)[0]?+?196.按行计算df的每一行均值
df[['col1','col2','col3']].mean(axis=1)97.对第二列计算移动平均值
#备注?每次移动三个位置,不可以使用自定义函数
np.convolve(df['col2'],?np.ones(3)/3,?mode='valid')98.将数据按照第三列值的大小升序排列
df.sort_values("col3",inplace=True)99.将第一列大于50的数字修改为'高'
df.col1[df['col1']?>?50]=?'高'100.计算第二列与第三列之间的欧式距离
np.linalg.norm(df['col2']-df['col3'])Part 5 一些补充
101.从CSV文件中读取指定数据
#?备注?从数据1中的前10行中读取positionName,?salary两列
df?=?pd.read_csv('数据1.csv',encoding='gbk',?usecols=['positionName',?'salary'],nrows?=?10)
df102.从CSV文件中读取指定数据
#?备注?从数据2中读取数据并在读取数据时将薪资大于10000的为改为高
df?=?pd.read_csv('数据2.csv',converters={'薪资水平':?lambda?x:?'高'?if?float(x)?>?10000?else?'低'}?)
df103.从上一题数据中,对薪资水平列每隔20行进行一次抽样
df.iloc[::20,?:][['薪资水平']]104.将数据取消使用科学计数法
#?输入
df?=?pd.DataFrame(np.random.random(10)**10,?columns=['data'])
df
df.round(3)105.将上一题的数据转换为百分数
df.style.format({'data':?'{0:.2%}'.format})106.查找上一题数据中第3大值的行号
df['data'].argsort()[::-1][7]107.反转df的行
df.iloc[::-1,?:]108.按照多列对数据进行合并
#?输入
df1=?pd.DataFrame({'key1':?['K0',?'K0',?'K1',?'K2'],
'key2':?['K0',?'K1',?'K0',?'K1'],
'A':?['A0',?'A1',?'A2',?'A3'],
'B':?['B0',?'B1',?'B2',?'B3']})
df2=?pd.DataFrame({'key1':?['K0',?'K1',?'K1',?'K2'],
'key2':?['K0',?'K0',?'K0',?'K0'],
'C':?['C0',?'C1',?'C2',?'C3'],
'D':?['D0',?'D1',?'D2',?'D3']})
pd.merge(df1,?df2,?on=['key1',?'key2'])109.按照多列对数据进行合并
pd.merge(df1,?df2,?how='left',?on=['key1',?'key2'])110.再次读取数据1并显示所有的列
df?=?pd.read_csv('数据1.csv',encoding='gbk')
pd.set_option("display.max.columns",?None)
df111.查找secondType与thirdType值相等的行号
np.where(df.secondType?==?df.thirdType)112.查找薪资大于平均薪资的第三个数据
np.argwhere(df['salary']?>?df['salary'].mean())[2]113.将上一题数据的salary列开根号
df[['salary']].apply(np.sqrt)114.将上一题数据的linestaion列按_拆分
df['split']?=?df['linestaion'].str.split('_')115.查看上一题数据中一共有多少列
df.shape[1]116.提取industryField列以'数据'开头的行
df[df['industryField'].str.startswith('数据')]117.按列制作数据透视表
pd.pivot_table(df,values=["salary","score"],index="positionId")118.同时对salary、score两列进行计算
df[["salary","score"]].agg([np.sum,np.mean,np.min])119.对salary求平均,对score列求和
df.agg({"salary":np.sum,"score":np.mean})120.计算并提取平均薪资最高的区
df[['district','salary']].groupby(by='district').mean().sort_values('salary',ascending=False).head(1)编辑:王菁
校对:林亦霖
—完—
想要获得更多数据科学领域相关动态,诚邀关注清华-青岛数据科学研究院官方微信公众平台“ 数据派THU ”。
