Cambiar la forma de los arrays

Parte de la sección Datos y gráficos del Journey de Machine Learning de Coddy. Lección 16 de 60.

Las ventas mensuales de una tienda, correspondientes a 12 meses, llegan en un único array plano. Agrúpalas en 4 trimestres de 3 meses y se abrirán nuevas posibilidades para plantear preguntas:

Una secuencia de 12 ventas mensuales, 5 8 6 9 12 10 14 15 11 7 6 9, se pliega en una cuadrícula de 4 filas etiquetadas de Q1 a Q4, tres meses por fila, en el mismo orden

.reshape(rows, columns) rellena las filas en orden:

months = np.array([5, 8, 6, 9, 12, 10, 14, 15, 11, 7, 6, 9])
quarters = months.reshape(4, 3)
print(quarters.sum(axis=1))
print(months.reshape(2, -1).shape)

Esto imprime:

[19 31 40 22]
(2, 6)

Los tamaños deben multiplicarse para obtener la cantidad: 4 por 3 es 12. Escribe -1 para uno de los tamaños y NumPy lo calcula.

.flatten() vuelve a convertir cualquier tabla en una sola matriz plana. Este capítulo termina con un juego de dados dividido en rondas de la misma manera.

Desafío

Fácil

Un lector registró sus minutos de lectura durante 3 semanas, de lunes a domingo, en el arreglo plano minutes. El programa lee un número de week: 1, 2 o 3.

Cambia la forma de minutes a weeks, con una fila por semana.
Imprime Weeks:, después la tabla.
Imprime el total de cada semana.
Imprime la fila de la semana elegida.

Para un registro de 2 semanas y la entrada 2:

Weeks:
[[10  0 25 30 15 45 20]
 [ 5 35 40  0 20 60 30]]
Weekly totals: [145 190]
Week 2: [ 5 35 40  0 20 60 30]

Pruébalo tú mismo

import numpy as np

# Minutos de lectura diarios durante 3 semanas, de lunes a domingo
minutes = np.array([20, 35, 15, 0, 40, 60, 45, 25, 30, 20, 10, 35, 50, 55,
                    30, 30, 25, 20, 45, 40, 70])
week = int(input())

# Escribe tu código a continuación