一千萬筆大約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
: 請問要怎麼修改可以速度快一點..
: 謝謝！