如何使用子字符串python查找特定字符串

Question

我和这个人有类似的问题：find position of a substring in a string

不同之处在于我不知道我的“mystr”是什么。我知道我的子字符串，但我在输入文件中的字符串可以是任意数量的随机单词，但我知道其中一个单词包括子字符串 cola。

例如 csv 文件：fanta,coca_cola,sprite 任意顺序。

如果我的子字符串是“cola”，那我怎么能写出这样的代码

mystr.find('cola')

或

match = re.search(r"[^a-zA-Z](cola)[^a-zA-Z]", mystr)

或

if "cola" in mystr

当我不知道我的“mystr”是什么时？

这是我的代码：

import csv

with open('first.csv', 'rb') as fp_in, open('second.csv', 'wb') as fp_out:
        reader = csv.DictReader(fp_in)
        rows = [row for row in reader]
        writer = csv.writer(fp_out, delimiter = ',')

        writer.writerow(["new_cola"])

        def headers1(name):
            if "cola" in name:
                    return row.get("cola")


        for row in rows:
                writer.writerow([headers1("cola")])

和第一个.csv：

fanta,cocacola,banana
0,1,0
1,2,1                      

打印出来

new_cola
""
""

什么时候打印出来

new_cola
1
2

Answer 1

这是一个工作示例：

import csv

with open("first.csv", "rb") as fp_in, open("second.csv", "wb") as fp_out:
        reader = csv.DictReader(fp_in)
        writer = csv.writer(fp_out, delimiter = ",")

        writer.writerow(["new_cola"])

        def filter_cola(row):
            for k,v in row.iteritems():
                if "cola" in k:
                    yield v

        for row in reader:
            writer.writerow(list(filter_cola(row)))

注意事项：

• rows = [row for row in reader] 是不必要且低效的（在这里您将生成器转换为列表，这会为大量数据消耗大量内存）
• 你的意思不是return row.get("cola")，而是return row.get(name)
• 在语句 return row.get("cola") 中，您访问当前范围之外的变量

• 您也可以使用 unix 工具cut。例如：

  cut -d "," -f 2 < first.csv > second.csv