函数中的Python子流程awk命令，将列表中的元素作为参数

Question

I want to subprocess an awk command with a list element as argument.

With a single argument, and from a shell prompt, it's prety simple:

$ awk -F "," '/US/ && /00001/ {print $1","$3}' stock_inventory.csv  > pretest_00001.csv

So using a list, I put it all in a Python script like:

import subprocess

mylist = [00001, 00002, 00003]

def myawk(item_code):
    subprocess.call("awk -F "," '/US/ && /%d/ {print $1","$3}' stock_inventory.csv  > pretest_%d.csv") % item_code

for i in mylist:
    myawk(i)

I did it wrong somewhere. Could Popen have been of any help? What about lambda in this case?

Thanks for you help.

Answer 1

perhaps ditch python and do all in awk ?

$ awk 'BEGIN{FS=OFS=",";
             n=split("00001,00002,00003",item)}
       /US/ {for(i=1;i<=n;i++) 
               if($0~item[i]) 
                 print $1,$3 > "pretest_"item[i]".csv"}' stock_inventory.csv  

Answer 2

Thanks all for your help.

I found a more elegant way to resolve it, though it's not through a Python script but a Bash one instead.

#!/bin/bash

declare -a arr=("00001" "00002" "00003")

for i in "${arr[@]}"

do

    cat stock_inventory.csv | grep $i | grep US | awk -F "," '{print $1","$3}' > pretest_$i.csv

done