Python MySQLDB: Ottieni il risultato di un risultato in un elenco

Question

Vorrei ottenere il risultato dell'operazione fetchall in un elenco invece di tuple di tuple o tuple di dizionari. Ad esempio,

cursor = connection.cursor() #Cursor could be a normal cursor or dict cursor 
query = "Select id from bs" 
cursor.execute(query) 
row = cursor.fetchall() 

Ora, il problema è la riga risultante è o ((123,), (234,)) o ({ 'id': 123}, { 'id': 234}) Quello che sto cercando è (123,234) o [123,234]. Siate i migliori se posso risparmiare parsing del resulset. Grazie in anticipo

Answer 1

E per quanto riguarda la comprensione delle liste? Se il risultato è ((123,), (234,), (345,)):

>>> row = [item[0] for item in cursor.fetchall()] 
>>> row 
[123, 234, 345] 

Se il risultato è ({'id': 123}, {'id': 234}, {'id': 345}):

>>> row = [item['id'] for item in cursor.fetchall()] 
>>> row 
[123, 234, 345] 

Answer 2

Se v'è un solo campo, posso usare questo per fare un elenco dal database:

def getFieldAsList(): 
    kursor.execute("Select id from bs") 
    id_data = kursor.fetchall() 
    id_list = [] 
    for index in range(len(id_data)): 
     id_list.append(id_data[index][0]) 
    return id_list 

Answer 3

cursor.execute("""Select * From bs WHERE (id = %s)""",(id)) 

cursor.fetchall() 

Answer 4

Sono sicuro che dopo tutto questo tempo, hai risolto questo problema, tuttavia, per alcune persone che potrebbero non sapere come ottenere i valori di un cursore come un dizionario utilizzando MySQLdb, è possibile utilizzare questo metodo trovato here:

import MySQLdb as mdb 

con = mdb.connect('localhost', 'testuser', 'test623', 'testdb') 

with con: 

    cur = con.cursor(mdb.cursors.DictCursor) 
    cur.execute("SELECT * FROM Writers LIMIT 4") 

    rows = cur.fetchall() 

    for row in rows: 
     print row["Id"], row["Name"] 

Answer 5

Questa vecchia Q arriva su Google durante la ricerca per l'appiattimento query db, per cui qui sono più suggerimenti. ..

Considerare un veloce list-flattening iterator.

Altre risposte utilizzano fetchall() che prima carica tutte le righe in memoria, quindi esegue un'iterazione per creare un nuovo elenco. Potrebbe essere inefficiente. Potrebbe combinano con MySQL cosiddetto server side cursor:

# assume mysql on localhost with db test and table bs 
import itertools 
import MySQLdb 
import MySQLdb.cursors 

conn = MySQLdb.connect(host='localhost',db='test', 
      cursorclass=MySQLdb.cursors.SSCursor) 
cursor = conn.cursor() 
# insert a bunch of rows 
cursor.executemany('INSERT INTO bs (id) VALUES (%s)',zip(range(1,10000))) 
conn.commit() 
# retrieve and listify 
cursor.execute("select id from bs") 
list_of_ids = list(itertools.chain.from_iterable(cursor)) 
len(list_of_ids) 
#9999 
conn.close() 

Ma la questione è anche etichettato Django, che ha una bella single field query flattener

class Bs(models.Model): 
    id_field = models.IntegerField() 

list_of_ids = Bs.objects.values_list('id_field', flat=True)