python - 将两个脚本合并为一个代码，用于 csv 文件数据验证

Question

大家好，目前我有两个脚本，我想将它们合并为 1 个代码。第一个脚本从一组数据中查找丢失的时间戳，并用 NaN 值填充空白行，然后保存到输出文件。第二个脚本比较一组数据中的不同行，并根据测试条件创建一个包含 True/False 值的新列。

如果我将每个脚本作为函数运行，然后使用另一个函数调用这两个脚本，我将得到两个单独的输出文件。如何仅使用 1 个已保存的输出文件来运行此操作？

第一个代码

import pandas as pd
df = pd.read_csv("data5.csv", index_col="DateTime", parse_dates=True)
df = df.resample('1min').mean()
df = df.reindex(pd.date_range(df.index.min(), df.index.max(), freq="1min"))
df.to_csv("output.csv", na_rep='NaN')

第二个代码

with open('data5.csv', 'r') as f:
    rows = [row.split(',') for row in f]
    rows = [[cell.strip() for cell in row if cell] for row in rows]

def isValidRow(row):
    return float(row[5]) <= 900 or all(float(val) > 7 for val in row[1:4])

header, rows = rows[0], rows[1:]
validRows = list(map(isValidRow, rows))

with open('output.csv', 'w') as f:
    f.write(','.join(header + ['IsValid']) + '\n')
    for row, valid in zip(rows, validRows):
        f.write(','.join(row + [str(valid)]) + '\n')

Answer 1

让您的代码作为文件名的函数:

def first_code(file_in, file_out):
    df = pd.read_csv(file_in, ... )
    ...
    df.to_csv(file_out, ...)

def second_code(file_in, file_out):
    with open(file_in, 'r') as f:
        ...
    ....

    with open(file_out, 'w') as f:
        ...

您的解决方案可以是:

first_code('data5.csv', 'output.csv')
second_code('output.csv', 'output.csv')

希望对你有帮助

<小时/>

注意，在同一个文件中读写没有问题。确保该文件之前已关闭以避免副作用。这是通过使用 with 隐式完成的，这是一个很好的实践