[英]Convert column in dataframe to “classes”?
所以我基本上得到了這個數據幀:
,club_name,tr_begin,year,ranking
0,ADO Den Haag,1357,2010,6.0
1,ADO Den Haag,1480,2011,15.0
2,ADO Den Haag,1397,2012,9.0
3,ADO Den Haag,1384,2013,9.0
4,ADO Den Haag,1451,2014,13.0
我想要做的就是這個,我想通過每個排名,並根據它的價值將它們放入一個類。 所以排名6將進入第2類,排名1將進入第1類。轉換表如下:
if ranking > 0 and ranking =< 3:
rank_class = 1
if ranking > 3 and ranking =< 6:
rank_class = 2
etc etc etc
我希望以3的倍數發生直到18。
所以我希望的輸出是:
,club_name,tr_begin,year,ranking, ranking_class
0,ADO Den Haag,1357,2010,6.0, 2
1,ADO Den Haag,1480,2011,15.0, 5
2,ADO Den Haag,1397,2012,9.0, 3
3,ADO Den Haag,1384,2013,9.0, 3
4,ADO Den Haag,1451,2014,13.0, 5
我嘗試使用掩碼功能,並通過創建一個新的數據幀然后合並,這工作,但似乎非常草率。 有一些簡單的方法來做到這一點?
提前致謝
使用pandas.cut
,您可以為“bin”和“labels”定義iterables。 這可以通過使用range
對象定義它們來簡化。
我建議你先將你的ranking
系列轉換為int
; 它可能受到浮點舍入的影響,這可能會產生不良結果。
df = pd.read_csv('file.csv')
binrange = range(0, 19, 3)
labrange = range(1, 7)
df['ranking_class'] = pd.cut(df['ranking'], bins=binrange, labels=labrange)
print(df)
club_name tr_begin year ranking ranking_class
0 ADO Den Haag 1357 2010 6.0 2
1 ADO Den Haag 1480 2011 15.0 5
2 ADO Den Haag 1397 2012 9.0 3
3 ADO Den Haag 1384 2013 9.0 3
4 ADO Den Haag 1451 2014 13.0 5
我認為整數除法//
會這樣做:
df.assign(ranking_class=(df.ranking // 3).astype(int))
club_name tr_begin year ranking ranking_class
0 ADO Den Haag 1357 2010 6.0 2
1 ADO Den Haag 1480 2011 15.0 5
2 ADO Den Haag 1397 2012 9.0 3
3 ADO Den Haag 1384 2013 9.0 3
4 ADO Den Haag 1451 2014 13.0 4
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.