Hay una cantidad de publicaciones que casi responden a esto pero o no las entiendo o no responden la pregunta:¿Cómo cambiar el tipo de un nucky recarray?
Tengo una recarray hecha usando numpy.rec.fromrecords. Supongamos que quiero convertir ciertas columnas en flotantes. ¿Cómo hago esto? ¿Debería cambiar a una ndarray y volver a una recarray?
Aquí hay un ejemplo usando astype para realizar la conversión:
import numpy as np
recs = [('Bill', '31', 260.0), ('Fred', 15, '145.0')]
r = np.rec.fromrecords(recs, formats = 'S30,i2,f4', names = 'name, age, weight')
print(r)
# [('Bill', 31, 260.0) ('Fred', 15, 145.0)]
El age es de dtype <i2:
print(r.dtype)
# [('name', '|S30'), ('age', '<i2'), ('weight', '<f4')]
podemos cambiar eso a <f4 usando astype:
r = r.astype([('name', '|S30'), ('age', '<f4'), ('weight', '<f4')])
print(r)
# [('Bill', 31.0, 260.0) ('Fred', 15.0, 145.0)]
Hay un re básicamente dos pasos. Mi obstáculo fue encontrar cómo modificar un dtype existente. Así es como lo hice:
# change dtype by making a whole new array
dt = data.dtype
dt = dt.descr # this is now a modifiable list, can't modify numpy.dtype
# change the type of the first col:
dt[0] = (dt[0][0], 'float64')
dt = numpy.dtype(dt)
# data = numpy.array(data, dtype=dt) # option 1
data = data.astype(dt)
¡Gracias! El "astype" es un poco más compacto que la recreación de una nueva matriz ... Supongo que es lo mismo en términos de eficiencia. Estoy publicando mi solución a continuación también, ya que incluye cómo modificar un tipo de letra existente. – mathtick