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

看板Python作者 (謬客)時間6年前 (2019/04/29 08:37), 編輯推噓0(000)
留言0則, 0人參與, 最新討論串2/4 (看更多)
一千萬筆大約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)》之銘言: : 想請教各位大大 : 在python裡面要怎麼寫出excel以下的功能: : if A2=A1, 則B2=A1+1 , 否則 B2=1 : 小弟試寫for+if如下, 但速度實在太慢, excel在B欄用if函數往下拉只需要5秒, : python卻要120秒... : df是dataframe, 有A, B兩欄 : for i in range(len(df)): : if df['A'][i+1] == df['A'][i]: : df['B'][i+1] = df['A'][i] + 1 : else: : df['B'][i+1] = 1 : 請問要怎麼修改可以速度快一點.. : 謝謝! -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.38.26.186 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1556498240.A.FBE.html
文章代碼(AID): #1SnaT0-- (Python)
討論串 (同標題文章)
文章代碼(AID): #1SnaT0-- (Python)