python-用分隔符在CSV文件中分割字符串

Question

我有一个包含以下数据的CSV文件：

Date,Profit/Losses
Jan-10,867884
Feb-10,984655
Mar-10,322013
Apr-10,-69417
May-10,310503
Jun-10,522857
Jul-10,1033096
Aug-10,604885
Sep-10,-216386
Oct-10,477532
Nov-10,893810
Dec-10,-80353

我已经将文件导入到python中，如下所示：

with open(csvpath, 'r', errors='ignore') as fileHandle:
lines = fileHandle.read()

我需要遍历这些行，以便仅提取“ Jan”，“ Feb”等月份，并将其放在其他列表中。 我还必须以某种方式跳过第一行，即Date, Profit/Losses （即标题）。

这是我到目前为止编写的代码：

months = []
for line in lines:
    months.append(line.split("-")

当我尝试打印months列表时，它会拆分文件中的每个字符！ 我在哪里错了？

Answer 1

您问题的一个答案是使用fileHandle.readlines() 。

lines = fileHandle.readlines()
# print(lines)
# ['Date,Profit/Losses\n', 'Jan-10,867884\n', 'Feb-10,984655\n', 'Mar-10,322013\n',
#  'Apr-10,-69417\n', 'May-10,310503\n', 'Jun-10,522857\n', 'Jul-10,1033096\n', 'Aug-10,604885\n',
#  'Sep-10,-216386\n', 'Oct-10,477532\n', 'Nov-10,893810\n', 'Dec-10,-80353\n']

for line in lines[1:]:
    # Starting from 2nd item in the list since you just want months
    months.append(line.split("-")[0])

Answer 2

您几乎可以始终使用专用工具（例如csv模块和列表理解）将痛苦降到最低：

import csv
with open("yourfile.csv") as infile:
    reader = csv.reader(infile) # Create a new reader
    next(reader) # Skip the first row
    months = [row[0].split("-")[0] for row in reader]

Answer 3

如果您确实想用困难的方法来尝试，请尝试以下操作：

months = []
for line in lines[1:]:
    months.append(line.split("-")[0])

lines[1:]将跳过第一行，而line.split("-")[0]将仅提取月份并追加到您的列表months 。

但是，正如AChampion所建议的那样，您应该真正研究一下csv或pandas软件包。

Answer 4

这样应该可以达到预期的效果（假设在同一目录中名为data.csv的文件）

result = []

with open('data.csv', 'r', encoding='UTF-8') as data:
    next(data)
    for record in data:
        result.append(record.split('-')[0])