2

我有一个 csv 文件，其中包含 3 行，如下所示：

Dallas
Houston
Ft. Worth

我想要做的是能够读取这些内容并从中制作链接，但将所有行输出在一行上。示例输出需要是这样的：

<a href="/dallas/">Dallas</a> <a href="/houston/">Houston</a> <a href="/ft-worth/">Ft. Worth</a>

这是我到目前为止的代码，它读取 csv 文件并输出，但它创建不同的行，我只想要一行，而且我需要在其中附加超链接的 html 代码。

f_in = open("data_files/states/major_cities.csv",'r')
for line in f_in.readlines():
    f_out.write(line.split(",")[0]+"")
f_in.close()
f_out.close()
4

2 回答 2

7

那是因为每个line输入f_in.readlines()都带有一个换行符。（尝试print(repr(line))在该循环中添加一个）。您需要做的是在writeing之前删除该换行符f_out：

for line in f_in.readlines():      
    actual_line = line.rstrip('\n')

您的整个代码如下所示：

import re 

with open('data_files/states/major_cities.csv') as f_in:
    with open('output_file.csv', 'w') as f_out:
        for line in f_in:
            city = line.rstrip('\n')
            f_out.write('<a href="/{}/">{}</a>'.format(
                re.sub(r'\W+', '-', city.lower()), 
                city
            ))

这些with语句负责close处理文件，因此您不需要最后两行。

更新

正如 JF Sebastian 指出的那样，还需要对城市名称进行slugify以实现您想要的输出。

于 2014-03-03T01:40:35.227 回答
2

尝试使用 python CSV 模块来处理 CSV 文件

import csv
file_out = open('file.txt','w')
with open('example.csv','rb') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        col=row[0]
        str="<a href=/" + col.strip().lower()
        str+= "/>" + col + "</a> "
        file_out.write(str)
于 2014-03-03T01:29:08.380 回答