numpy: czy to możliwe, aby zachować dtype kolumn przy użyciu column_stack

Question

Gdy używam column_stack aby złączyć NumPy tablicami dtype zostaje zamienione (zazwyczaj unosić)

a = numpy.array([1., 2., 3.], dtype=numpy.float64) 
b = numpy.array([1, 2, 3], dtype=numpy.int64) 
print numpy.column_stack((a, b)).dtype 
>>> float64 

czy istnieje sposób, aby zachować dtype z poszczególne kolumny?

Answer 1

można układać dwie tablice z numpy.lib.recfunctions metody i zachowania typu z nim:

>>> from numpy.lib.recfunctions import append_fields 

>>> a = numpy.rec.array(a, dtype=[('a', numpy.float64)]) 
>>> new_a = append_fields(a, 'b', b, usemask=False, dtypes=[numpy.int64]) 
>>> new_a 
array([(1.0, 1), (2.0, 2), (3.0, 3)], 
     dtype=[('a', '<f8'), ('b', '<i8')]) 

>>> new_a['a'] 
array([ 1., 2., 3.]) 

>>> new_a['b'] 
array([1, 2, 3]) 

Answer 2

Moje tablice został przekształcony w ciąg (S18), kiedy column_stack'ed em.

Użyłem astype (pożądany dtype), aby przekonwertować je z powrotem do tego, co było po układaniu.

http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.astype.html

przykład:

new_array= old_array.astype(float64)