尝试从DataFrame的第一个非零值开始添加时出现问题。列中间有零值的记录

import pandas as pd

def add_data_from_first_nonzero(df, data):
    for col in df.columns:
        first_nonzero_index = df[col].ne(0).idxmax()
        df[col] = df[col].shift(-first_nonzero_index)
        df.loc[first_nonzero_index:, col] = data

# 示例用法
df = pd.DataFrame({'A': [0, 0, 1, 2, 0, 3],
                   'B': [0, 4, 5, 0, 6, 7]})
data_to_add = [10, 20, 30, 40, 50, 60]

add_data_from_first_nonzero(df, data_to_add)
print(df)

import pandas as pd

def add_data_from_first_nonzero(col, data):
    first_nonzero_index = col.ne(0).idxmax()
    col = col.shift(-first_nonzero_index)
    col.loc[first_nonzero_index:] = data
    return col

# 示例用法
df = pd.DataFrame({'A': [0, 0, 1, 2, 0, 3],
                   'B': [0, 4, 5, 0, 6, 7]})
data_to_add = [10, 20, 30, 40, 50, 60]

df = df.apply(add_data_from_first_nonzero, args=(data_to_add,))
print(df)