Я пытаюсь получить данные из MySQL db в пару массивов numpy. Код выглядит следующим образом:
import numpy as np
import mysql.connector
mydb = mysql.connector.connect(
[redacted]
)
mycursor = mydb.cursor()
mycursor.execute("select foo,bar,baz from mytable order by id")
features = np.asarray(mycursor.fetchall())
print(features[0:5])
mycursor.execute("select gik from mytable order by id")
codes = np.asarray(mycursor.fetchall())
print(codes)
mydb.close()
features
превращается в массив ранга 2, чего я и хочу. Однако codes
также является массивом ранга 2, но мне нужен массив ранга 1 (т.е. одномерный). Я просмотрел документацию и обнаружил, что для этого есть метод flatten, но я попытался добавить codes.flatten()
прямо перед оператором печати, и это не сработало. Может ли кто-нибудь указать мне правильное направление?