python - 从查询结果中获取字符串的一部分

时间:2016-04-27 11:44:30

标签: postgresql python-2.7

我有一个小脚本,可以从查询中下载结果并将其导出到.csv

yy = raw_input("Enter year: ")
mm = raw_input("Enter month: ")

try:

    conn = psycopg2.connect("dbname='xxx' user='xxx' host='xxx' password='xxx'")
    cur=conn.cursor()

 query1 = """SELECT name FROM xxx where name like '%{0}_{1}%' """.format(yy,mm)
        cur.execute(query1)
        results = cur.fetchall()
        for row in results:
        print row

csv_file = open('errors.csv','wb')# wb - query importing line by line
myFile = csv.writer(csv_file)
myFile.writerows(results)
csv_file.close()

我需要的是查询结果中字符串的一部分。 Atm我的结果如下:

1-GRB-804_2016_02_03__08_42_12
1-GRB-804_2016_02_05__11_04_47
1-GRB-804_2016_02_06__08_20_15
1-GRB-804_2016_02_08__08_06_13
1-GRB-804_2016_02_08__08_30_58

我想要的是获得第一线直到它符合标志' _'。 结果将是:

1-GRB-804
1-GRB-804
1-GRB-804
1-GRB-804
1-GRB-804

此致

1 个答案:

答案 0 :(得分:1)

你想要保留所有内容直到第一个“_”,你可以简单地使用string.split

>>> s = '1-GRB-804_2016_02_03__08_42_12'
>>> s.split('_')
['1-GRB-804', '2016', '02', '03', '', '08', '42', '12']
>>> s = s.split('_')[0]
>>>> s
'1-GRB-804'

所以在你的情况下,你需要这样做:

for row in results:
    print row.split('_')[0]