Question

需要在现有的excel文件表中添加一列，并使用python panda或任何其他更好的库在整个列中应用此excel公式，并将其写入到现有的excel文件中。 excel公式为：

=IF(AND(J2="SC",J3="EUS),K3-K2,"")

或

=IF(AND(currentRowOfCol J="SC",NextRowOfCol J="EUS),NextRowOfCol K-CurrentRowOfCol K,"")

示例：

结果应为：

index J     K   L
  1   SC    4   1  
  2   EUS   5  NULL
  3   SC    2  NULL

尝试过此代码：

import openpyxl

wb = openpyxl.load_workbook('C:/Users/Sameer/Desktop/abc.xlsx')
Sheet = wb['Data Owner Group History']

for cellObj in list(Sheet.columns)[12]:
#    print(n)
    j1=('J'+str(cellObj.row))
    j2 = ('J'+str((cellObj.row)+1))
    y=('K'+(str((cellObj.row)+1)))
    k=('K2'+(str(cellObj.row)))
    n='IF(AND({0}="SC",{1}="EUS-CUC-DISPATCH (Ceryx/Amdocs group)"),{2}- 
    {3},"")'.format(j1,j2,y,k)
    cellObj.value = n
    wb.save("C:/Users/Sameer/Desktop/result.xlsx")
print("done")

但是它卡在执行中，并且从不执行最后一个命令print（“ done”）。它会创建一个结果文件，但已损坏并且不包含任何内容。如果有人可以使用熊猫来实现它会更好，那么操作会更快。

预先感谢

Answer 1

IIUC np.where和shift

np.where((df.J.eq('SC'))&(df.J.shift(-1).eq('EUS')),df.K.shift(-1)-df.K,None)
Out[7]: array([1.0, None, None], dtype=object)

df['L']=np.where((df.J.eq('SC'))&(df.J.shift(-1).eq('EUS')),df.K.shift(-1)-df.K,None)
df
Out[9]: 
   index    J  K     L
0      1   SC  4     1
1      2  EUS  5  None
2      3   SC  2  None

在工作表中添加一列，然后使用python pandas或任何其他更好的库在整个列中应用此excel公式

1 个答案: