感谢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