Python：在文件中循环查找特定行

Question

我要在第三列中的文件中有以下几行： 在文件中，我没有数字列：

1. 红色; 蓝色; 绿色; 白色; 橙子;
2. 绿色; 白色; 橙子;
3. 蓝色; 绿色; 白色;
4. 红色; 蓝色; 绿色; 白色;
5. 蓝色; 绿色; 白色; 橙子;
6. 橙子
7. 绿色; 白色; 橙子;
8. 白色; 橙子
9. 绿色;

我使用以下代码行来做到这一点：

lines = i.split(";")[2]

问题是某些行只有一两列，所以它给我“索引超出范围”错误。 请告诉我如何解决这个问题？

非常感谢Adia

Answer 1

那这样的事情呢：

cols = i.split(";")
if (len(cols) >= 3):
    lines = cols[2]
else:
    #whatever you want here

Answer 2

简单的解决方案是检查列数，并忽略少于三列的行。

third_columns = []
with open("...") as infile:
    for line in infile:
        columns = line.split(';')
        if len(columns) >= 3:
            third_columns.append(columns[2])

而且，如果您解析CSV（似乎与您一样），则最好使用众多现有CSV解析器之一， 例如，标准库中的之一 。

Answer 3

使用切片而不是索引。

>>> with open('test.txt') as f_in:
...     column3 = (line.split(';')[2:3] for line in f_in)
...     column3 = [item[0] for item in column3 if item]
... 
>>> column3
[' Green', ' Orange', ' White', ' Green', ' White', ' Orange']

Answer 4

for line in open("file"):
    try:
        s=line.split(";")[2]
    except: pass
    else:
        print s