想请教各位大大
在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
请问要怎么修改可以速度快一点..
谢谢！

改用numpy array, 有需要再转字典

题外话，尽量别用chain index的方式取值..基本上df尽量不要用for去操作，可以使用内建method就用，速度差蛮多的

np.where 可以试试

因为你用index 要速度差很多

谢谢大家！！努力研究如何避免在df中用for操作..速度真的差很多！！

不只是pandas的dataframe，其实一般能不用for就不用for..

不用for 怎么循环执行？？

所以我是说“能不用就不用啊”，像有限次的矩阵运算这种能够用现成package运算塞满pipe line来提升效率是最好～