import pandas as pd
def append_columns(input_file, new_data):
# 读取原始CSV文件
df_original = pd.read_csv(input_file)
# 创建新的DataFrame
df_new_data = pd.DataFrame(new_data, columns=['New Column 1', 'New Column 2'])
# 将新数据的前两列追加到原始数据的接下来两列
df_original.iloc[:, -2:] = df_new_data
# 显示更新后的DataFrame
print(df_original)
if __name__ == "__main__":
input_file = "input.csv"
new_data = [(1, 2), (3, 4), (5, 6)] # 这里是新读取的数据
append_columns(input_file, new_data)
这段代码假设有一个名为 "input.csv" 的 CSV 文件,包含至少四列数据。新读取的数据被传递为一个列表,其中每个元素是一个包含两个值的元组,表示新数据的两列。
代码首先读取原始 CSV 文件,然后将新数据创建为一个 Pandas 数据帧,并将其两列追加到原始数据的最后两列。最后,更新后的数据帧被打印出来。可以根据需要对数据帧进行保存或执行其他操作。