excel - numpy

# -*- coding: utf-8 -*-
"""
Created on Thu May 28 07:56:51 2015

@author: aluno
"""

import numpy as np

my_array = np.array([1,2,3,4])
print my_array.dtype

print len(my_array)

str_array = my_array.astype(str)
float_array = str_array.astype(float)

for i in my_array:
    print i*4

dim2_array = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
print dim2_array.ndim
print dim2_array.shape
print dim2_array.size

# cálculo da média
print np.mean(dim2_array)
# média por coluna
print np.mean(dim2_array,axis=0)
# média por linha
print np.mean(dim2_array,axis=1)

# outra maneira de fazer media
# de todos os elementos
media = 0
(linhas,colunas) = dim2_array.shape
for i in xrange(0,linhas):
    for j in xrange(0,colunas):
        media = media + dim2_array[i,j]

media = float(media)/dim2_array.size
print "media da array =",media

# outra maneira de fazer media
# por coluna

(linhas,colunas) = dim2_array.shape
for j in xrange(0,colunas):
    media = 0
    for i in xrange(0,linhas):
        media = media + dim2_array[i,j]
    media = float(media)/linhas
    print "media da coluna =", media

tabela = np.genfromtxt('planilha2.csv',delimiter=";",skiprows=1)

print tabela[0,0]
print tabela[:,0]
print tabela[0,:]

# média de cada turma
print tabela[:,0].mean()
print tabela[:,1].mean()
print tabela[:,2].mean()
print tabela[:,3].mean()

# em um único comando
print tabela.mean(axis=0)
(linhas,colunhas) = tabela.shape
medias = tabela.mean(axis=0)

for j in xrange(0,colunas):
    print "Turma",j+1," = ",medias[j]

tabela = tabela + 1

np.savetxt('saida.csv',tabela,fmt="%.1f",delimiter=";")