dataframe基本操作

行列数

返回列数：
df.shape[1]
返回行数：
df.shape[0]
或者：
len(df)

索引

切片

高级操作

apply 和 applymap

df.apply(func)
取df中的每一列作为func的输入
即运行：func(df[0])，func(df[1])，func(df[2])。。。。返回这些值形成的一个series

def calculate_dihedrals_from_df(series,pos):
    idx4 = series.values
    
    print(pos)
    p0 = pos[idx4][0]
    p1 = pos[idx4][1]
    p2 = pos[idx4][2]
    p3 = pos[idx4][3]
    
    # print(p0,p1,p2,p3)
    
    DihedralAngle = getDihedralAngle(p0,p1,p2,p3)
    return DihedralAngle

pairs_atoms4.apply(calculate_dihedrals_from_df,pos=pos)
# pairs_atoms4是一个dataframe