Python 复制 CSV 数据并添加下一行数据
这是一段实现上述要求的 Python 代码:
import pandas as pd
# 读取csv文件
df = pd.read_csv('input.csv')
# 复制每一行为7行
df = pd.concat([df]*7, ignore_index=True)
# 添加新列
df['next_row'] = pd.Series([df.iloc[i+7]['col1'] for i in range(len(df)-7)] + [None]*7)
# 输出结果
print(df)
这段代码首先使用 Pandas 的 read_csv 函数读取输入的 CSV 文件,并将其存储为一个 DataFrame 对象。然后使用 concat 函数将每一行复制为 7 行,并使用 ignore_index=True 参数重置索引,以便后续操作。接着,使用 pd.Series 函数创建一个新的 Series 对象,其中包含了原数据下一行的 7 个数字。通过在列表推导式中使用 iloc 函数,可以遍历每一行数据,并取出其下一行的数据。最后,将这个新的 Series 对象赋值给新增的列 'next_row',并输出结果。
原文地址: https://www.cveoy.top/t/topic/nwaJ 著作权归作者所有。请勿转载和采集!