python - 将数组添加到 numpy 数组

Question

我想要一个看起来像这样的 numpy 数组:

X = np.array([[10, 20], [20, 25], [30, 16], [40, 18], [50, 90], [60, 87]])

我目前有从 firestore 检索的字典值:

doc_ref = db.collection('CPU Logs')
query_ref = doc_ref.where(u'testData', u'==', True).order_by(u'logId')
docs = query_ref.get()

我遍历它们并将键值分配给 2 个变量，id 和 usage，然后将它们添加到数组 toAppend:

for doc in docs:
    values = doc.to_dict()
    id = values['logId']
    usage = values['usage']
    toAppend = [id, usage]

toAppend 看起来像 [10, 30] 如果 id 是 10 并且使用是 30。现在，我在尝试将它添加到一个空的 numpy 数组时遇到了问题。我试过插入:

X = np.array([])
for doc in docs:
    values = doc.to_dict()
    id = values['logId']
    usage = values['usage']
    toAppend = [id, usage]
    a = X.flatten()
    np.insert(a, [0,0], toAppend)

print(X)

以及追加:

np.append(X, toAppend)

但两者似乎都不起作用，因为 print 语句只是打印出 []。

Answer 1

看看the docs for insert和 flatten :他们都返回新数组(副本)。所以你需要写

X = np.insert(a, [0, 0], toAppend)

为了让 X 包含扩展数组。我也不认为您需要前面的 X.flatten()。

与其插入一个 numpy 数组(这很昂贵)，您应该考虑构建一个嵌套的 list 并且只在最后转换一次。