Re: [問題] for + if 優化問題

作者: Brahms5566 (Brahms)   2019-04-29 23:50:06
感謝TuCH大大的code! 成功了超快!
但又出現一個問題..如果說原本code的以下標記^的部分
改成df['B'][i] + 1 請問要怎麼寫會比較好呢?
就像excel中, 符合條件的欄位, 則B2=B1+1, 如果B1=1, 則B2 = 2
然後往下拉, 依序是3, 4, 5...直到下一個不合條件的值為原本的1,
接著再2, 3, 4..持續下去
謝謝!
: : for i in range(len(df)):
: if df['A'][i+1] == df['A'][i]:
: df['B'][i+1] = df['A'][i] + 1
^^^^^^^^^^
改成=>df['B'][i] + 1
: else:
: df['B'][i+1] = 1
※ 引述《TuCH (謬客)》之銘言:
: 一千萬筆大約0.6秒 給您參考看看
: import pandas as pd
: import numpy as np
: # 隨機產生測試資料
: data = np.random.randint(1,9,10000000)
: # pandas 讀取
: df = pd.DataFrame(data)
: # column 取名
: df.columns = ['A']
: # column B 初始化
: df['B'] = 1
: # 計算
: df.loc[df['A'].diff()==0,'B'] = df['A'].shift()+1
作者: Brahms5566 (Brahms)   2019-04-30 21:18:00
剛剛嘗試用group然後for enumerate轉numpy順利得到想要的結果了~謝謝大家指引迷津
作者: BlgAtlfans (BLG_Eric)   2019-04-30 12:40:00
其實全部換成list操作比較好 盡量不要用pandas跑迴圈
作者: TuCH (謬客)   2019-04-30 08:17:00
這個每一個值都跟前一個值有關係 看來要用迴圈了
作者: sherees (ShaunTheSheep)   2019-04-30 11:10:00
如果條件判斷不會隨著B欄的值更動產生變化 用前面Tu大的寫法應該就可以了
作者: jp956956 (儲備里長伯)   2019-05-02 03:06:00
list操作什麼意思?? 先list做完再給pandas處理嗎?

Links booklink

Contact Us: admin [ a t ] ucptt.com