不改变格式的浮点数组中的字符串数组

Question

有没有办法在浮点数数组中包含字符串而不改变数组的格式，这样所有浮点数都更改为字符串但字符串元素仍保留为字符串？

例如。

import numpy as np

a = np.array([ 'hi' , 1. , 2. , 3. ])

理想情况下，我希望格式保持与上面输入为“a”时的外观相同。

这给出了：

array(['hi', '1.0', '2.0', '3.0'], dtype='|S3')

那么如何将这样的数组保存为文本文件呢？

非常感谢，

J

Answer 1

我猜您的问题是这样的：您想使用 np.savetxt() 转储数组 np.array([ 'hi' , 1. , 2. , 3. ])，但出现此错误：

TypeError: Mismatch between array dtype ('|S3') and format specifier ('%.18e')

如果是这种情况，您只需要在np.savetxt 中设置fmt kwarg。除了用于格式化浮点数据的默认%.18e，您可以使用%s，它将事物格式化为字符串，即使数组中的原始值是数字。

所以这会起作用：

import numpy as np
a = np.array([ 'hi' , 1. , 2. , 3. ])
np.savetxt("test.out",a,fmt="%s")

请注意，您可以使用原始列表执行此操作 - numpy 会为您将其转换为数组。所以例如你可以这样做：

np.savetxt("test.out",[ 'hi' , 1. , 2. , 3. ],fmt="%s")

它也应该可以正常工作。

对于问题的第一部分，这并不是 numpy 数组的真正用途。如果您尝试将不同的数据类型放入同一个数组中，那么您可能需要不同的数据结构。一个普通的 python 列表就可以了，但根据你的情况，dict 可能是你要找的。​​p>

编辑：根据评论线程和特定问题，看起来这是在数据文件上创建标题的尝试。这可以直接通过

np.savetxt("a.txt",a,header="title goes here")

这可以直接使用np.loadtxt() 读取，因为默认情况下，标题以# 开头，默认情况下np.loadtxt() 忽略以# 开头的行。

Answer 2

使用泡菜：

import pickle

a = ['abc',3,4,5,6,7.0]
pickle.dump( a, open( "save.p", "wb" ))
b = pickle.load( open( "save.p", "rb" ) )

print(b)

输出：

['abc', 3, 4, 5, 6, 7.0]