根据值将成对转换为具有不同行数的列表

Question

我有一组成对的起点和终点数据作为行。其中每一个都有从起点移动到目的地的人数。这个人数变化很大（从约 4 人到 >200 人）。

我想要实现的是将这些行转换为列表方式，其中两行代表一个移动......这听起来很不直观，但这是我需要将数据读入映射包的格式正在使用。

我希望下面的图片能更好地说明我正在尝试做的事情 - 红色单元格代表 4 次移动（总计列），然后将其转换为 8 行，起点和终点配对重复 4 次。

Excel 将在此期间工作（作为概念证明），但如果我要处理整个数据集，我将获得超过 200 万条记录（至少） - 所以数据库或 Python 解决方案将需要 - 只是为了生成最终列表的 *.csv 文件。

注意 - Total 列加倍 表示需要的行数 - 所以如果 5 个人移动，则需要 10 行。

所以 - 我已将此问题标记为 Excel、Access 或 Python 解决方案。我是 VBA 或 Python 的新手，但我愿意接受建议。

更新 非常感谢萨罗！该解决方案效果很好（适当更改了输入和输出 .csv 的文件路径） - 这是从数据中提取的动画的屏幕截图！

Answer 1

我会假设您的输入如下所示：

origx, origy, destx, desty, Total
0.0, 0.0, 1.1, 1.1, 2
151.1556, -33.9113, 150.9991, -33.7297, 4

将其转换为：

ID, X, Y, Column pair
1, 0.0, 0.0, origx|origy
1, 1.1, 1.1, destx|desty
2, 0.0, 0.0, origx|origy
2, 1.1, 1.1, destx|desty
3, 151.1556, -33.9113, origx|origy
3, 150.9991, -33.7297, destx|desty
4, 151.1556, -33.9113, origx|origy
4, 150.9991, -33.7297, destx|desty
5, 151.1556, -33.9113, origx|origy
5, 150.9991, -33.7297, destx|desty
6, 151.1556, -33.9113, origx|origy
6, 150.9991, -33.7297, destx|desty

Python 代码

def write_header(f):
    f.write('ID, X, Y, Column pair\n')

def skip_header(f):
    f.next()

def main():
    my_id = 1
    with open('input.csv', 'r') as in_f:
        skip_header(in_f)
        with open('output.csv', 'w') as out_f:
            write_header(out_f)
            for line in in_f:
                orig_x, orig_y, dest_x, dest_y, total = \
                    [x.strip() for x in line.split(',')]
                for idx in range(int(total)):
                    out_f.write(
                        '{}, {}, {}, origx|origy\n'.format(my_id, orig_x, orig_y))
                    out_f.write(
                        '{}, {}, {}, destx|desty\n'.format(my_id, dest_x, dest_y))
                    my_id += 1


if __name__ == '__main__':
    main()

我没有在代码中写很多cmets，所以如果您需要，请要求更多说明:)