擅长:python、mysql、java
<p>给出一个完全人为的csv,如下所示(另存为'测试.csv'):</p>
<pre><code>firstitem, seconditem, thirditem, fourthitem, fifthitem
first, price, 1, nothing, important
second, price, 2, over, here
</code></pre>
<p>然后,以下内容将提取包含整数的所有列:</p>
<pre><code>>>> def get_items(pattern, csv_file):
with open(csv_file) as file:
for entry in csv.DictReader(file):
for field_name, value in entry.items():
if re.search(pattern, value):
yield [field_name, value]
>>> data = list(get_items(r'\d+', 'test.csv'))
[[' thirditem', ' 1'], [' thirditem', ' 2']]
</code></pre>
<p>或者,可以使用<code>if value.strip().isdigit()</code>作为条件语句,而不必使用regex。你知道吗</p>