我只是在玩数字游戏。
我发现Numpy提供了一个名为np.vectorize的函数,允许您获取一个函数并将其应用于Numpy数组。
在[23]中:import numpy as np
import decimal
D = decimal.Decimal
我们将创建一个常规的np浮点数组
在[24]中:f10 = np.random.ranf(10)
f10
出[24]:array([ 0.45410583, 0.35353919, 0.5976785 , 0.12030978, 0.00976334,
0.47035594, 0.76010096, 0.09229687, 0.24842551, 0.30564141])
尝试使用np.asarray将数组转换为十进制类型不起作用。似乎尝试使用np.asarray并指定decimal.decimal类型会将数组设置为预期的object,但如果实际访问数组的单个元素,则它仍然具有float数据类型。
在[25]中:f10todec = np.asarray(f10, dtype = decimal.Decimal)
print f10todec.dtype, f10todec
print type(f10todec[0])
object [0.454105831376884 0.3535391906233327 0.5976785016396975 0.1203097778312584
0.009763339031407026 0.47035593879363524 0.7601009625324361
0.09229687387940333 0.24842550566826282 0.30564141425653435]
如果给np.array一个十进制类型的同质python列表,则它似乎保留了该类型,请参阅下面的列表理解,以获得第一个数组中作为十进制数据类型的值的列表。所以我不得不这样做十进制数组。
在[26]中:D10 = np.array([D(d) for d in f10])
D10
出[26]:array([Decimal('0.4541058313768839838076019077561795711517333984375'),
Decimal('0.35353919062333272194109667907468974590301513671875'),
Decimal('0.597678501639697490332991947070695459842681884765625'),
Decimal('0.12030977783125840208100498784915544092655181884765625'),
Decimal('0.00976333903140702563661079693702049553394317626953125'),
Decimal('0.47035593879363524205672320022131316363811492919921875'),
Decimal('0.76010096253243608632743644193396903574466705322265625'),
Decimal('0.09229687387940332943259136300184763967990875244140625'),
Decimal('0.24842550566826282487653543284977786242961883544921875'),
Decimal('0.30564141425653434946951847450691275298595428466796875')], dtype=object)
基本的数学运算似乎没问题
在[27]中:D10/2
出[27]:array([Decimal('0.2270529156884419919038009539'),
Decimal('0.1767695953116663609705483395'),
Decimal('0.2988392508198487451664959735'),
Decimal('0.06015488891562920104050249392'),
Decimal('0.004881669515703512818305398469'),
Decimal('0.2351779693968176210283616001'),
Decimal('0.3800504812662180431637182210'),
Decimal('0.04614843693970166471629568150'),
Decimal('0.1242127528341314124382677164'),
Decimal('0.1528207071282671747347592373')], dtype=object)
在[28]中:np.sqrt(D10)
出[28]:array([Decimal('0.6738737503248542354573624759'),
Decimal('0.5945916166776426405934196108'),
Decimal('0.7730966961769384578392278689'),
Decimal('0.3468569991095154505863255680'),
Decimal('0.09880961001545864636229121433'),
Decimal('0.6858250059553349663476168402'),
Decimal('0.8718376927688066448819998853'),
Decimal('0.3038040057000620415496242404'),
Decimal('0.4984230187985531079935481296'),
Decimal('0.5528484550548498633920483390')], dtype=object)
在尝试十进制模块中没有对应函数的trig函数之前
在[29]中:np.sin(D10)
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
in ()
----> 1 np.sin(D10)
AttributeError: 'Decimal' object has no attribute 'sin'
所以让我们使用np.vectorize,这样我们就可以使用decimal的量化函数来进行舍入。In [30]:
npquantize = np.vectorize(decimal.Decimal.quantize)
qnt_D10 = npquantize(D10, D('.000001'))
qnt_D10
Out[30]:
array([Decimal('0.454106'), Decimal('0.353539'), Decimal('0.597679'),
Decimal('0.120310'), Decimal('0.009763'), Decimal('0.470356'),
Decimal('0.760101'), Decimal('0.092297'), Decimal('0.248426'),
Decimal('0.305641')], dtype=object)
您还需要注意一些常规的python数学函数,因为它们会自动将返回类型更改为float。我假设这是因为这个数不能基于SIN或COS这样的函数来精确计算。
所以我猜简单的答案是使用列表理解来获取numpy数组中的项并将其转换为python列表,然后从小数列表中创建该数组。
要返回类型完整的numpy数组,我想您可以使用vectorize函数包装任何使用Decimal类型的函数以应用于np数组。