python - 如何使用 csv_writer 将列表写入单独的行？

Question

我用了BeautifulSoup提取 html 表并将元素存储在列表中。接下来，我想将列表写入 .csv 文件，但看起来 writerow 函数不会将元素写入不同的行。

import csv
from bs4 import BeautifulSoup

# Grab first table (station table in html file)
def parse_station(html):
    soup = BeautifulSoup(html)
    s_table = soup.find_all('table')[1]
    stations = []
    for tr in s_table.find_all('tr')[1:]:
        td = tr.find_all('td')[1]
        td = td.get_text()
        stations.append(td)

    return stations


stations = parse_station(open('data.html').read())

with open('stations.csv', "wb") as f:
    csv_writer = csv.writer(f)
    csv_writer.writerow([stations])
f.close()

.csv 就像:

A,B,C,D,E

而不是:

A,
B,
C,
D,
E,

我的代码有什么问题吗？我该如何修复它？ (我使用的是Python 2.7)

Answer 1

您可以使用此示例代码

import csv
with open('test.csv', "wb") as f:
    writer = csv.writer(f)
    writer.writerow(['A'])
    writer.writerow(['B'])

这会给你这样的结果

A
B

你可以传递你的值(value)

注意:检查电台类型，如果这将返回 str，则您的值将在单行中，但如果这是列表，则循环遍历列表示例代码以将列表写入 CSV。

>>> list = [1,2,3,4,5]
>>> with open('test.csv', 'wb') as f:
...     writer = csv.writer(f)
...     for i in list:
...         writer.writerow([i])
...