DataAnalysis/Wc.py at master · rodolfojbrandao/DataAnalysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import os
import numpy as np
import pandas as pd
import math
from pandas import DataFrame

NF = 683 #number of files
NP = 10500 #number of particles

vX = np.zeros((NP,NF))
vY = np.zeros((NP,NF))
vZ = np.zeros((NP,NF))

path='/home/rodolfo/Documents/regimes/45/dados'
os.chdir(path)

for j in range (0,NF):
    Dados = pd.read_csv('dados.{}.csv'.format(j))
    concluido=j/NF*100
    print(concluido)
    print(j)
    vX[:, j] = Dados.iloc[:, 2].copy()
    vY[:, j] = Dados.iloc[:, 3].copy()
    vZ[:, j] = Dados.iloc[:, 4].copy()

vX=pd.DataFrame(vX)
vY=pd.DataFrame(vY)
vZ=pd.DataFrame(vZ)

v=np.sqrt(np.square(vX)+np.square(vY)+np.square(vZ))

V=v.mean(axis = 0)
V  = pd.DataFrame(V)
V.to_csv('Velocidade45.csv')
print(type(V))