python - 导出至 CSV 文件

Question

我正在尝试将此代码的结果导出到 CSV 文件。我在代码后面复制了 2 个结果。每只股票有 14 种商品，我想写入一个 CSV 文件，并为 14 种商品中的每一种提供一列，为每只股票提供一行。

import requests
from bs4 import BeautifulSoup

base_url = 'http://finviz.com/screener.ashx?v=152&s=ta_topgainers&o=price&c=0,1,2,3,4,5,6,7,25,63,64,65,66,67'
html = requests.get(base_url)
soup = BeautifulSoup(html.content, "html.parser")
main_div = soup.find('div', attrs = {'id':'screener-content'})
table = main_div.find('table')
sub = table.findAll('tr')
rows = sub[5].findAll('td')
for row in rows:
    link = row.a
    if link is not None:
        print(link.get_text())

这是结果的格式，每只股票 14 个项目/列。

PTN
Palatin Technologies, Inc.
Healthcare
Diagnostic Substances
USA
240.46M
9.22
193.43M
2.23M
0.76
1.19
7.21%
1,703,285
3
LKM
Link Motion Inc.
Technology
Application Software
China
128.95M
-
50.40M
616.76K
1.73
1.30
16.07%
1,068,798
4

尝试过但无法使其工作。

TextWriter x = File.OpenWrite ("my.csv", ....);

x.WriteLine("Column1,Column2"); // header
x.WriteLine(coups.Cells[0].Text + "," + coups.Cells[1].Text);

Answer 1

这应该有效:

import requests
from bs4 import BeautifulSoup

base_url = 'http://finviz.com/screener.ashx?v=152&s=ta_topgainers&o=price&c=0,1,2,3,4,5,6,7,25,63,64,65,66,67'
html = requests.get(base_url)
soup = BeautifulSoup(html.content, "html.parser")
main_div = soup.find('div', attrs = {'id':'screener-content'})

light_rows = main_div.find_all('tr', class_="table-light-row-cp")
dark_rows = main_div.find_all('tr', class_="table-dark-row-cp")

data = []
for rows_set in (light_rows, dark_rows):
    for row in rows_set:
        row_data = []
        for cell in row.find_all('td'):
            val = cell.a.get_text()
            row_data.append(val)
        data.append(row_data)

#   sort rows to maintain original order
data.sort(key=lambda x: int(x[0]))

import pandas
pandas.DataFrame(data).to_csv("AAA.csv", header=False)

这里有一些事情:

1. 我使用了“table-[light|dark]-row-cp”，因为所有感兴趣的行都有这些类之一(并且没有其他行有它们)
2. 有两个独立的部分:一个是以正确的结构获取数据，另一个 - 写入 CSV 文件。
3. 我使用 pandas CSV writer，因为我熟悉它，但是当您有矩形数据(此处称为“数据”)时，您可以使用任何其他 CSV writer
4. 切勿使用保留名称来命名变量，例如“sub”或“link”:)

希望有帮助。