欄目今天介紹pandas的操作。
pandas有一種功能非常強大的方法,它就是accessor,可以將它理解為一種屬性介面,通過它可以獲得額外的方法。其實這樣說還是很籠統,下面我們通過程式碼和範例來理解一下。
>>> pd.Series._accessors {'cat', 'str', 'dt'}複製程式碼
對於Series資料結構使用_accessors方法,我們得到了3個物件:cat,str,dt。
下面我們依次看一下這三個物件是如何使用的。
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個方法說明:
其實不難發現,該用法的使用與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個方法說明:
這個用法就有點複雜了,因為很明顯看到,這是一個鏈式的用法。通過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']複製程式碼
屬性有很多,對於具體的用法,如果感興趣可以自己進行摸索練習。
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種方法說明:
其它方法也都是基於datetime的一些變換,並通過變換來檢視具體微觀或者宏觀日期。
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其它相關文章!