数据规整化
清理
转换
合并
重塑
数据库风格的DataFrame合并
pd.merge(df1, df2) # 默认会将重叠列的列名当作键,最好显式的指定下,另外merge默认是使用的inner join pd.merge(df1, df2, on='key') pd.merge(df3, df4, left_on='lkey', right_on='rkey') # 如果两个对象的列名不同,需要分别指定,如上df3里面的是lkey,df4里面的是rkey pd.merge(df1, df2, how='outer') # how参数用来指定连接方式,总共是inner、left、right、outer四种方式 pd.merge(df1, df2, on='key', how='left') pd.merge(df1, df2, how='inner') # 多对多合并如此简单 pd.merge(left, right, on=['key1', 'key2'], how='outer') # 多个合并列的时候,列表 pd.merge(left, right, on='key1') # 需要注意的是如果以key1进行连接,但是两个对象里面有其他相同列名的列存在, left会被表示成key_x,right会被表示成key_y pd.merge(left, right, on='key1', suffixes=('_left', '_right')) # 可以自定义的为左右设定后缀,这样相当于定制了列名
索引上的合并
pd.merge(left1, right1, left_on='key', right_index=True) # 左边是的key,右边的是index pd.merge(left1, right1, left_on='key', right_index=True, how='outer') # 合并方式跟之前一样还是inner、outer、left、right pd.merge(lefth, righth, left_on=['key1', 'key2'], right_index=True) # 合并键是多个列用列表指明 pd.merge(lefth, righth, left_on=['key1', 'key2'], right_index=True, how='outer') # 同样可以设置合并方式 pd.merge(left2, right2, how='outer', left_index=True, right_index=True) # 两个合并对象都通过index连接 left2.join(right2, how='outer') # dataframe里面提供了join方法,用来更方便的实现按索引合并,不过join支持的是左连接 left1.join(right1, on='key') # 还支持参数dataframe的索引跟调用dataframe的列进行连接 left2.join([right2, another]) left2.join([right2, another], how='outer') # 对于简单的索引合并,你还可以向join传入一组DataFrame
轴向连接
刚刚上面讲了数据层的横向连接合并,现在是关于数据堆叠。NumPy的concatenation函数可以用NumPy数组来做:
arr = np.arange(12).reshape((3, 4)) np.concatenate([arr, arr], axis=1) # 默认axis=0,axis为1的时候就会变成横向的拼接
而在pandas里面提供了concat函数
s1 = pd.Series([0, 1], index=['a', 'b']) s2 = pd.Series([2, 3, 4], index=['c', 'd', 'e']) s3 = pd.Series([5, 6], index=['f', 'g']) pd.concat([s1, s2, s3]) # concat可以将值和索引粘合在一起
pd.concat([s1, s2, s3], axis=1) # concat是在axis=0上工作的,最终产生一个新的Series。如果传入axis=1,则结果就会变成一个DataFrame(axis=1是列) pd.concat([s1, s2, s3], axis=1)
pd.concat([s1, s4], axis=1, join='inner') # join='inner' 可以得到两个对象的交集 pd.concat([s1, s4], axis=1, join_axes=[['a', 'c', 'b', 'e']]) # 你可以通过join_axes指定要在其它轴上使用的索引 # 会建立在另外一个索引上 result = pd.concat([s1, s1, s3], keys=['one','two', 'three']) result.unstack() #有个问题,参与连接的片段在结果中区分不开。假设你想要在连接轴上创建一个层次化索引。使用keys参数即可达到这个目的
pd.concat([s1, s2, s3], axis=1, keys=['one','two', 'three']) # 如果沿着axis=1对Series进行合并,则keys就会成为DataFrame的列头
df1 = pd.DataFrame(np.arange(6).reshape(3, 2), index=['a', 'b', 'c'], columns=['one', 'two']) df2 = pd.DataFrame(5 + np.arange(4).reshape(2, 2), index=['a', 'c'],columns=['three', 'four']) pd.concat([df1, df2], axis=1, keys=['level1', 'level2']) # 在dataframe里面也是同理 pd.concat({'level1': df1, 'level2': df2}, axis=1) # 如果传入的不是列表而是字典,则健值就变成keys pd.concat([df1, df2], axis=1, keys=['level1', 'level2'], names=['upper', 'lower']) 此外还有两个用于管理层次化索引创建方式的参数(参见表8-3)。举个例子,我们可以用names参数命名创建的轴级别 df1 = pd.DataFrame(np.random.randn(3, 4), columns=['a', 'b', 'c', 'd']) df2 = pd.DataFrame(np.random.randn(2, 3), columns=['b', 'd', 'a']) pd.concat([df1, df2], ignore_index=True) # DataFrame的行索引不包含任何相关数据,在这种情况下,传入ignore_index=True即可
合并重复数据
还有一种数据组合问题不能用简单的合并(merge)或连接(concatenation)运算来处理。比如说,你可能有索引全部或部分重叠的两个数据集。举个有启发性的例子,我们使用NumPy的where函数,它表示一种等价于面向数组的if-else:
a = pd.Series([np.nan, 2.5, np.nan, 3.5, 4.5, np.nan], index=['f', 'e', 'd', 'c', 'b', 'a']) b = pd.Series(np.arange(len(a), dtype=np.float64), index=['f', 'e', 'd', 'c', 'b', 'a']) b[-1] = np.nan np.where(pd.isnull(a), b, a) # array([ 0. , 2.5, 2. , 3.5, 4.5, nan]) where(if a then a else b) b[:-2].combine_first(a[2:]) # 以b为标准,不存在的索引用a内的索引 a NaN b 4.5 c 3.0 d 2.0 e 1.0 f 0.0
对于dataframe,也是同样的
df1 = pd.DataFrame({'a': [1., np.nan, 5., np.nan], 'b': [np.nan, 2., np.nan, 6.], 'c': range(2, 18, 4)}) df2 = pd.DataFrame({'a': [5., 4., np.nan, 3., 7.], 'b': [np.nan, 3., 4., 6., 8.]}) df1.combine_first(df2)
重塑层次化索引
- stack:将数据的列“旋转”为行。
- unstack:将数据的行“旋转”为列。
data = pd.DataFrame(np.arange(6).reshape((2, 3)), index=pd.Index(['Ohio','Colorado'], name='state'),columns=pd.Index(['one', 'two', 'three'],name='number')) result = data.stack() # 对该数据使用stack方法即可将列转换为行,得到一个Series result.unstack() # 对于一个层次化索引的Series,你可以用unstack将其重排为一个DataFrame
result.unstack(0) result.unstack('state') # 默认情况下,unstack操作的是最内层(stack也是如此)。传入分层级别的编号或名称即可对其它级别进行unstack操作 s1 = pd.Series([0, 1, 2, 3], index=['a', 'b', 'c', 'd']) s2 = pd.Series([4, 5, 6], index=['c', 'd', 'e']) data2 = pd.concat([s1, s2], keys=['one', 'two']) data2.unstack() # 如果不是所有的级别值都能在各分组中找到的话,则unstack操作可能会引入缺失数据 data2.unstack().stack() # stack默认会滤除缺失数据,因此该运算是可逆的 data2.unstack().stack(dropna=False) # 可以通过参数来取消滤缺失数据
df = pd.DataFrame({'left': result, 'right': result + 5}, columns=pd.Index(['left', 'right'], name='side')) df.unstack('state') # 在对DataFrame进行unstack操作时,作为旋转轴的级别将会成为结果中的最低级别 df.unstack('state').stack('side') # 当调用stack,我们可以指明轴的名字
'长格式'旋转为'宽格式'
pivoted = data.pivot('date', 'item', 'value') # pivot内(行索引, 列索引, 值) pivoted = data.pivot('date', 'item') #如果有两个同时需要重塑的数据列,忽略最后一个参数,得到的DataFrame就会带有层次化的列 unstacked = ldata.set_index(['date', 'item']).unstack('item') # pivot其实就是用set_index创建层次化索引,再用unstack重塑
将'宽格式'旋转为'长格式'
旋转DataFrame的逆运算是pandas.melt。它不是将一列转换到多个新的DataFrame,而是合并多个列成为一个,产生一个比输入长的DataFrame
df = pd.DataFrame({'key': ['foo', 'bar', 'baz'], 'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}) melted = pd.melt(df, ['key']) # key列可能是分组指标,其它的列是数据值。当使用pandas.melt,我们必须指明哪些列是分组指标。
reshaped = melted.pivot('key', 'variable', 'value') # 使用pivot,可以重塑回原来的样子 reshaped.reset_index() # 因为pivot的结果从列创建了一个索引,用作行标签,我们可以使用reset_index将数据移回列 pd.melt(df, id_vars=['key'], value_vars=['A', 'B']) # 还可以指定列的子集,作为值的列 pd.melt(df, value_vars=['A', 'B', 'C']) pd.melt(df, value_vars=['key', 'A', 'B']) # pandas.melt也可以不用分组指标
移除重复数据
DataFrame的duplicated方法返回一个布尔型Series,表示各行是否是重复行(前面出现过的行)
data = pd.DataFrame({'k1': ['one', 'two'] * 3 + ['two'], 'k2': [1, 1, 2, 3, 3, 4, 4]}) data.duplicated() data.drop_duplicates() # drop_duplicates重复的列移除掉,返回dataframe,移除的列是后出现的重复列,即上面的值为True的
这两个方法默认会判断全部列,你也可以指定部分列进行重复项判断。假设我们还有一列值,且只希望根据k1列过滤重复项
data['v1'] = range(7) data.drop_duplicates(['k1'])
data.drop_duplicates(['k1', 'k2'], keep='last') # duplicated和drop_duplicates默认保留的是第一个出现的值组合。传入keep='last'则保留最后一个
利用函数或映射进行数据转换
data = pd.DataFrame({'food': ['bacon', 'pulled pork', 'bacon','Pastrami', 'corned beef', 'Bacon','pastrami', 'honey ham', 'nova lox'] ,'ounces': [4, 3, 12, 6, 7.5, 8, 3, 5, 6]})
假设你想要添加一列表示该肉类食物来源的动物类型。我们先编写一个不同肉类到动物的映射:
meat_to_animal = { 'bacon': 'pig', 'pulled pork': 'pig', 'pastrami': 'cow', 'corned beef': 'cow', 'honey ham': 'pig', 'nova lox': 'salmon' }
Series的map方法可以接受一个函数或含有映射关系的字典型对象,但是这里有一个小问题,即有些肉类的首字母大写了,而另一些则没有。因此,我们还需要使用Series的str.lower方法,将各个值转换为小写
data['animal'] = data['food'].str.lower().map(meat_to_animal)
也可以传入一个能够完成全部这些工作的函数
data['food'].map(lambda x: meat_to_animal[x.lower()])
替换值
data = pd.Series([1., -999., 2., -999., -1000., 3.]) data.replace(-999, np.nan) # 将-999替换成NA值 data.replace([-999, -1000], np.nan) # 将多个值转换成NA值 data.replace([-999, -1000], [np.nan, 0]) #要让每个值有不同的替换值,可以传递一个替换列表 data.replace({-999: np.nan, -1000: 0}) # 传入的参数也可以是字典
笔记:data.replace方法与data.str.replace不同,后者做的是字符串的元素级替换。
重命名轴索引
data = pd.DataFrame(np.arange(12).reshape((3, 4)), index=['Ohio', 'Colorado', 'New York'], columns=['one', 'two', 'three', 'four']) transform = lambda x: x[:4].upper() data.index.map(transform) data.index = data.index.map(transform) # 修改index
如果想要创建数据集的转换版(而不是修改原始数据),比较实用的方法是rename
data.rename(index=str.title, columns=str.upper)
rename可以结合字典型对象实现对部分轴标签的更新:
data.rename(index={'OHIO': 'INDIANA'}, columns={'three': 'peekaboo'})
rename可以实现复制DataFrame并对其索引和列标签进行赋值。如果希望就地修改某个数据集,传入inplace=True即可
data.rename(index={'OHIO': 'INDIANA'}, inplace=True)
离散化和面元划分
为了便于分析,连续数据常常被离散化或拆分为面元。
ages = [20, 22, 25, 27, 21, 23, 37, 31, 61, 45, 41, 32] bins = [18, 25, 35, 60, 100] cats = pd.cut(ages, bins) # 将age列表数据划分到面元,相当于归类到区间
cats.codes # array([0, 0, 0, 1, 0, 0, 2, 1, 3, 2, 2, 1], dtype=int8) cats.categories # IntervalIndex([(18, 25], (25, 35], (35, 60], (60, 100]], closed='right', dtype='interval[int64]') pd.value_counts(cats)
跟“区间”的数学符号一样,圆括号表示开端,而方括号则表示闭端(包括)。哪边是闭端可以通过right=False进行修改
cats = pd.cut(ages, bins, right=False)
你可 以通过传递一个列表或数组到labels,设置自己的面元名称:
group_names = ['Youth', 'YoungAdult', 'MiddleAged', 'Senior'] pd.cut(ages, bins, labels=group_names)
如果向cut传入的是面元的数量而不是确切的面元边界,则它会根据数据的最小值和最大值计算等长面元。下面这个例子中,我们将一些均匀分布的数据分成四组:
data = np.random.rand(20) pd.cut(data, 4, precision=2) # 选项precision=2,限定小数只有两位。
qcut是一个非常类似于cut的函数,它可以根据样本分位数对数据进行面元划分。根据数据的分布情况,cut可能无法使各个面元中含有相同数量的数据点。而qcut由于使用的是样本分位数,因此可以得到大小基本相等的面元:
data = np.random.randn(1000) cats = pd.qcut(data, 4) pd.value_counts(cats) # 可以发现四个区间是一样大的都是250个元素
与cut类似,你也可以传递自定义的分位数(0到1之间的数值,包含端点):
pd.qcut(data, [0, 0.1, 0.5, 0.9, 1.])