python - 如何在python中将选定的数据从一列移动到另一列

Question

我正在读取一个包含两列的 CSV 文件:大洲、国家/地区。每个大陆下面的单元格都是空的。

Continent    |  Countries
-----------------------------
    Asia       |   China
               |   Japan
               |   Korea
    North Am.  |   Canada
               |   U.S.

但我想将这两列合并到 Continent 列下并进行缩进。

Continent
-----------
Asia 
    China
    Japan
    Korea
North America
    Canada
    U.S.

我如何使用 python 执行此操作？

Answer 1

下面的代码应该可以解决问题。你甚至不需要使用 pandas 来读取 CSV，因为 Python 已经有一个名为 csv 的标准模块。 .代码输出到另一个名为 output.csv 的 CSV。

import csv

continent = ''
with open('data.csv') as csv_in, open('output.csv', 'w') as csv_out:
    reader = csv.reader(csv_in)
    writer = csv.writer(csv_out)
    headers = next(reader, None) # skip the headers
    writer.writerow([headers[0]])
    for row in reader:
        prev_continent = continent
        continent = row[0].strip() or continent

        if prev_continent != continent:
          writer.writerow([continent])
        writer.writerow([f'    {row[1]}'])

输出.csv

Continent
Asia
    China
    Japan
    Korea
North Am.
    Canada
    U.S.

数据.csv

Continent,Countries
Asia     ,China
         ,Japan
         ,Korea
North Am.,Canada
         ,U.S.