如何在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.