Código Rápido para Análise de Dados #03
- #Data
- #Python
Diante da matriz de dados:
[ [1, 2], [3, 4], [5, 6]]
Escreva um código em Python usando a biblioteca NumPy que crie essa matriz e calcule a soma dos valores de cada coluna separadamente.
# Entrada - Python
import numpy as np
data = np.array([
[1, 2],
[3, 4],
[5, 6]
])
result = data.sum(axis=0)
print(result)
# Saída
[ 9 12]
# Código comentado
import numpy as np
# Importa a biblioteca NumPy como np
data = np.array([[1, 2], [3, 4], [5, 6]])
# Cria um array NumPy bidimensional (3x2) com os seguintes valores:
# [[1, 2],
# [3, 4],
# [5, 6]]
result = data.sum(axis=0)
# O parâmetro axis define o eixo que a soma deve ocorrer: axis=0: soma as colunas; axis=1: soma as linhas.
# Calcula a soma dos elementos ao longo do eixo 0 (ou seja, soma as colunas).
# Isso significa somar os valores linha por linha nas mesmas colunas:
# Coluna 1: 1 + 3 + 5 = 9
# Coluna 2: 2 + 4 + 6 = 12
# Resultado: array([ 9, 12])
print(result)
# Imprime a saída
Fonte: Código extraído de questão de concurso.
