python - 运行后memmap文件在哪里？

Question

我有一个使用 numpy memmap 的矩阵乘法代码

def test_memmap():
    fA = np.lib.format.open_memmap('A.npy', dtype='uint8', mode='r+')
    fB = np.lib.format.open_memmap('B.npy', dtype='uint8', mode='r+')

    #need to predefine size of result?
    #fC = np.memmap('C.npy', dtype='uint16', mode='w+', shape=(rows,rows))

    # print fA
    # print fB

    #where it stored if not gived filename?

    t0= time.time()
    fC= np.dot(fA,fB)
    print (time.time()-t0)

    print fC.filename
    print type(fC)

    # print fC[12:10]
    # print fC

运行后memmap文件fC在哪？以及如何指定保存路径？

我如何指向保存 fC 的位置(而不是将其存储在内存中)？可以自动检测数组的类型和大小吗？

Answer 1

虽然fC是numpy.core.memmap.memmap的一个实例，它不与任何文件关联:

print(type(fC))
# <class 'numpy.core.memmap.memmap'>

print(fC._mmap)
# None

print(fC.filename)
# None

fC 是 memmap 实例的原因是因为 np.dot 与大多数 NumPy 函数一样，试图返回与其参数类型相同的数组。所以 memmap.__array_wrap__ method被调用。 __array_finalize__ method最后调用它，将 _mmap、filename、offset 和 mode 属性设置为 None。

如果您使用像 pdb 这样的调试器单步执行代码，您可以看到 __array_finalize__ 被调用。

所以虽然fC是一个memmap实例，但它只存在于内存中。

---

计算 np.dot 并将结果写入 memmap:

fC = np.memmap('/tmp/C.npy', dtype='float', mode='w+', shape=(Arows, Bcols))
np.dot(fA,fB, out=fC)
print fC.filename
# /tmp/C.npy
print type(fC)
# <class 'numpy.core.memmap.memmap'>