1%的人知道的pandas騷操作,傳授給你

2020-10-09 18:00:27

欄目今天介紹pandas的操作。

pandas有一種功能非常強大的方法,它就是accessor,可以將它理解為一種屬性介面,通過它可以獲得額外的方法。其實這樣說還是很籠統,下面我們通過程式碼和範例來理解一下。

>>> pd.Series._accessors
{'cat', 'str', 'dt'}複製程式碼

對於Series資料結構使用_accessors方法,我們得到了3個物件:cat,str,dt

  • .cat:用於分類資料(Categorical data)
  • .str:用於字元資料(String Object data)
  • .dt:用於時間資料(datetime-like data)

下面我們依次看一下這三個物件是如何使用的。

str物件的使用

Series資料型別:str字串

# 定義一個Series序列
>>> addr = pd.Series([
...     'Washington, D.C. 20003',
...     'Brooklyn, NY 11211-1755',
...     'Omaha, NE 68154',
...     'Pittsburgh, PA 15211'
... ]) 

>>> addr.str.upper()
0     WASHINGTON, D.C. 20003
1    BROOKLYN, NY 11211-1755
2            OMAHA, NE 68154
3       PITTSBURGH, PA 15211
dtype: object

>>> addr.str.count(r'\d') 
0    5
1    9
2    5
3    5
dtype: int64複製程式碼

關於以上str物件的2個方法說明:

  • Series.str.upper:將Series中所有字串變為大寫;
  • Series.str.count:對Series中所有字串的個數進行計數;

其實不難發現,該用法的使用與Python中字串的操作很相似。沒錯,在pandas中你一樣可以這樣簡單的操作,而不同的是你操作的是一整列的字串資料。仍然基於以上資料集,再看它的另一個操作:

>>> regex = (r'(?P<city>[A-Za-z ]+), '      # 一個或更多字母
...          r'(?P<state>[A-Z]{2}) '        # 兩個大寫字母
...          r'(?P<zip>\d{5}(?:-\d{4})?)')  # 可選的4個延伸數位
...
>>> addr.str.replace('.', '').str.extract(regex)
         city state         zip
0  Washington    DC       20003
1    Brooklyn    NY  11211-1755
2       Omaha    NE       68154
3  Pittsburgh    PA       15211複製程式碼

關於以上str物件的2個方法說明:

  • Series.str.replace:將Series中指定字串替換;
  • Series.str.extract:通過正規表示式提取字串中的資料資訊;

這個用法就有點複雜了,因為很明顯看到,這是一個鏈式的用法。通過replace將 " . " 替換為"",即為空,緊接著又使用了3個正規表示式(分別對應city,state,zip)通過extract對資料進行了提取,並由原來的Series資料結構變為了DataFrame資料結構。

當然,除了以上用法外,常用的屬性和方法還有.rstrip,.contains,split等,我們通過下面程式碼檢視一下str屬性的完整列表:

>>> [i for i in dir(pd.Series.str) if not i.startswith('_')]
['capitalize',
 'cat',
 'center',
 'contains',
 'count',
 'decode',
 'encode',
 'endswith',
 'extract',
 'extractall',
 'find',
 'findall',
 'get',
 'get_dummies',
 'index',
 'isalnum',
 'isalpha',
 'isdecimal',
 'isdigit',
 'islower',
 'isnumeric',
 'isspace',
 'istitle',
 'isupper',
 'join',
 'len',
 'ljust',
 'lower',
 'lstrip',
 'match',
 'normalize',
 'pad',
 'partition',
 'repeat',
 'replace',
 'rfind',
 'rindex',
 'rjust',
 'rpartition',
 'rsplit',
 'rstrip',
 'slice',
 'slice_replace',
 'split',
 'startswith',
 'strip',
 'swapcase',
 'title',
 'translate',
 'upper',
 'wrap',
 'zfill']複製程式碼

屬性有很多,對於具體的用法,如果感興趣可以自己進行摸索練習。

dt物件的使用

Series資料型別:datetime

因為資料需要datetime型別,所以下面使用pandas的date_range()生成了一組日期datetime演示如何進行dt物件操作。

>>> daterng = pd.Series(pd.date_range('2017', periods=9, freq='Q'))
>>> daterng
0   2017-03-31
1   2017-06-30
2   2017-09-30
3   2017-12-31
4   2018-03-31
5   2018-06-30
6   2018-09-30
7   2018-12-31
8   2019-03-31
dtype: datetime64[ns]

>>>  daterng.dt.day_name()
0      Friday
1      Friday
2    Saturday
3      Sunday
4    Saturday
5    Saturday
6      Sunday
7      Monday
8      Sunday
dtype: object

>>> # 檢視下半年
>>> daterng[daterng.dt.quarter > 2]
2   2017-09-30
3   2017-12-31
6   2018-09-30
7   2018-12-31
dtype: datetime64[ns]

>>> daterng[daterng.dt.is_year_end]
3   2017-12-31
7   2018-12-31
dtype: datetime64[ns]複製程式碼

以上關於dt的3種方法說明:

  • Series.dt.day_name():從日期判斷出所處星期數;
  • Series.dt.quarter:從日期判斷所處季節;
  • Series.dt.is_year_end:從日期判斷是否處在年底;

其它方法也都是基於datetime的一些變換,並通過變換來檢視具體微觀或者宏觀日期。

cat物件的使用

Series資料型別:Category

在說cat物件的使用前,先說一下Category這個資料型別,它的作用很強大。雖然我們沒有經常性的在記憶體中執行上g的資料,但是我們也總會遇到執行幾行程式碼會等待很久的情況。使用Category資料的一個好處就是:可以很好的節省在時間和空間的消耗。下面我們通過幾個範例來學習一下。

>>> colors = pd.Series([
...     'periwinkle',
...     'mint green',
...     'burnt orange',
...     'periwinkle',
...     'burnt orange',
...     'rose',
...     'rose',
...     'mint green',
...     'rose',
...     'navy'
... ])
...
>>> import sys
>>> colors.apply(sys.getsizeof)
0    59
1    59
2    61
3    59
4    61
5    53
6    53
7    59
8    53
9    53
dtype: int64複製程式碼

上面我們通過使用sys.getsizeof來顯示記憶體佔用的情況,數位代表位元組數。
還有另一種計算內容佔用的方法:memory_usage(),後面會使用。

現在我們將上面colors的不重複值對映為一組整數,然後再看一下佔用的記憶體。

>>> mapper = {v: k for k, v in enumerate(colors.unique())}
>>> mapper
{'periwinkle': 0, 'mint green': 1, 'burnt orange': 2, 'rose': 3, 'navy': 4}

>>> as_int = colors.map(mapper)
>>> as_int
0    0
1    1
2    2
3    0
4    2
5    3
6    3
7    1
8    3
9    4
dtype: int64

>>> as_int.apply(sys.getsizeof)
0    24
1    28
2    28
3    24
4    28
5    28
6    28
7    28
8    28
9    28
dtype: int64複製程式碼

注:對於以上的整數值對映也可以使用更簡單的pd.factorize()方法代替。

我們發現上面所佔用的記憶體是使用object型別時的一半。其實,這種情況就類似於Category data型別內部的原理。

記憶體佔用區別:Categorical所佔用的記憶體與Categorical分類的數量和資料的長度成正比,相反,object所佔用的記憶體則是一個常數乘以資料的長度。

下面是object記憶體使用和category記憶體使用的情況對比。

>>> colors.memory_usage(index=False, deep=True)
650
>>> colors.astype('category').memory_usage(index=False, deep=True)
495複製程式碼

上面結果是使用object和Category兩種情況下記憶體的佔用情況。我們發現效果並沒有我們想象中的那麼好。但是注意Category記憶體是成比例的,如果資料集的資料量很大,但不重複分類(unique)值很少的情況下,那麼Category的記憶體佔用可以節省達到10倍以上,比如下面資料量增大的情況:

>>> manycolors = colors.repeat(10)
>>> len(manycolors) / manycolors.nunique() 
20.0

>>> manycolors.memory_usage(index=False, deep=True)
6500
>>> manycolors.astype('category').memory_usage(index=False, deep=True)
585複製程式碼

可以看到,在資料量增加10倍以後,使用Category所佔內容節省了10倍以上。

除了佔用記憶體節省外,另一個額外的好處是計算效率有了很大的提升。因為對於Category型別的Series,str字元的操作發生在.cat.categories的非重複值上,而並非原Series上的所有元素上。也就是說對於每個非重複值都只做一次操作,然後再向與非重複值同類的值對映過去。

對於Category的資料型別,可以使用accessor的cat物件,以及相應的屬性和方法來操作Category資料。

>>> ccolors = colors.astype('category')
>>> ccolors.cat.categories
Index(['burnt orange', 'mint green', 'navy', 'periwinkle', 'rose'], dtype='object')複製程式碼

實際上,對於開始的整數型別對映,我們可以先通過reorder_categories進行重新排序,然後再使用cat.codes來實現對整數的對映,來達到同樣的效果。

>>> ccolors.cat.reorder_categories(mapper).cat.codes
0    0
1    1
2    2
3    0
4    2
5    3
6    3
7    1
8    3
9    4
dtype: int8複製程式碼

dtype型別是Numpy的int8(-127~128)。可以看出以上只需要一個單位元組就可以在記憶體中包含所有的值。我們開始的做法預設使用了int64型別,然而通過pandas的使用可以很智慧的將Category資料型別變為最小的型別。

讓我們來看一下cat還有什麼其它的屬性和方法可以使用。下面cat的這些屬性基本都是關於檢視和操作Category資料型別的。

>>> [i for i in dir(ccolors.cat) if not i.startswith('_')]
['add_categories',
 'as_ordered',
 'as_unordered',
 'categories',
 'codes',
 'ordered',
 'remove_categories',
 'remove_unused_categories',
 'rename_categories',
 'reorder_categories',
 'set_categories']複製程式碼

但是Category資料的使用不是很靈活。例如,插入一個之前沒有的值,首先需要將這個值新增到.categories的容器中,然後再新增值。

>>> ccolors.iloc[5] = 'a new color'
# ...
ValueError: Cannot setitem on a Categorical with a new category,
set the categories first

>>> ccolors = ccolors.cat.add_categories(['a new color'])
>>> ccolors.iloc[5] = 'a new color'  
複製程式碼

如果你想設定值或重塑資料,而非進行新的運算操作,那麼Category型別不是那麼有用。

相關免費學習推薦:(視訊)

以上就是1%的人知道的pandas騷操作,傳授給你的詳細內容,更多請關注TW511.COM其它相關文章!