DataAnalysis/MixingIndexSerial.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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import os
import numpy as np
import pandas as pd
import math

#NF = 4 #number of files
NF = 2 #number of files
NP = 147465 #number of particles
raio1 = 0.0009
raio2 = 0.0021
rx = 5
ry = 5
rz = 10
tamanho = rx*ry*rz

velocidade_resultante = pd.DataFrame(np.zeros((NP,NF)))
contador1 = list()
contador2 = list()
contador = list()
np.linspace(0.1,49.0,NF)
posicaoX = np.zeros((NP,NF))
posicaoY = np.zeros((NP,NF))
posicaoZ = np.zeros((NP,NF))
Tipo = np.zeros((NP,NF))
Raio = np.zeros((NP,NF))
vx = np.zeros((NP,NF))
vy = np.zeros((NP,NF))
vz = np.zeros((NP,NF))
contador_tipo1 = pd.DataFrame(np.zeros((tamanho,NF)))
contador_tipo2 = pd.DataFrame(np.zeros((tamanho,NF)))
velocidade_tipo1 = pd.DataFrame(np.zeros((tamanho,NF)))
velocidade_tipo2 = pd.DataFrame(np.zeros((tamanho,NF)))
posicao_vetor = 1

path='/home/rodolfo/Desktop/dados'
os.chdir(path)

# leitura e separaçao de dados
for j in range (0,NF):
    Dados = pd.read_csv('dados.{}.csv'.format(j))
    posicaoX[:, j] = Dados.iloc[:, 6].copy()
    posicaoY[:,j] = Dados.iloc[:, 7].copy()
    posicaoZ[:,j] = Dados.iloc[:, 8].copy()
    Tipo[:,j] = Dados.iloc[:, 1].copy()
    Raio[:,j] = Dados.iloc[:, 5].copy()
    vx[:,j] = Dados.iloc[:, 2].copy()
    vy[:,j] = Dados.iloc[:, 3].copy()
    vz[:,j] = Dados.iloc[:, 4].copy()

posicaoX=pd.DataFrame(posicaoX)
posicaoY=pd.DataFrame(posicaoY)
posicaoZ=pd.DataFrame(posicaoZ)
Tipo=pd.DataFrame(Tipo)
Raio=pd.DataFrame(Raio)
vx=pd.DataFrame(vx)
vy=pd.DataFrame(vy)
vz=pd.DataFrame(vz)

gradeamentoX = (max(posicaoX) - min(posicaoX)) / rx
gradeamentoY = (max(posicaoY) - min(posicaoY)) / ry
gradeamentoZ = (max(posicaoZ) - min(posicaoZ)) / rz

for i in range(0,NF):
    velocidade_resultante.iloc[:,[i]] = vx.iloc[:,[i]]*vx.iloc[:,[i]]+vy.iloc[:,[i]]*vy.iloc[:,[i]]+vz.iloc[:,[i]]*vz.iloc[:,[i]]

velocidade_resultante=velocidade_resultante.pow(1./2)
V_medio = velocidade_resultante.mean(axis=0)

InferiorX=float(posicaoX.iloc[:,[1]].min())
InferiorY=float(posicaoY.iloc[:,[1]].min())
InferiorZ=float(posicaoZ.iloc[:,[1]].min())

for m in range(0,NF):
    p=0
    q=0
    concluido = m/NF*100
    print(concluido)
    for n in range(1, 100):
        posicaoXX = posicaoX.iloc[n,m].copy()
        posicaoYY = posicaoY.iloc[n,m].copy()
        posicaoZZ = posicaoZ.iloc[n,m].copy()
        #print('ok')
        for pz in range(1,rz):
            for py in range(1,ry):
                for px in range(1,rx):
                    if posicaoXX>InferiorX+(px-1)*gradeamentoX and posicaoXX<InferiorX+px*gradeamentoX \
                            and posicaoYY>InferiorY+(py-1)*gradeamentoY and posicaoYY<InferiorY+py*gradeamentoY \
                            and posicaoZZ>InferiorZ+(pz-1)*gradeamentoZ and posicaoZZ<InferiorZ+pz*gradeamentoZ:
                        posicao_vetor = -31 + px + 5 * py + 25 * pz

                        if Tipo.iloc[n,m]<1.5:
                            contador_tipo1.iloc[posicao_vetor,m] = contador_tipo1.iloc[posicao_vetor,m]+1
#                            p=p+1
#                            velocidade_tipo1.iloc[p,m] = velocidade_resultante.iloc[n,m].copy()
                        else:
                            contador_tipo2.iloc[posicao_vetor, m] = contador_tipo2.iloc[posicao_vetor, m] + 1
#                            q = q + 1
#                            velocidade_tipo2.iloc[p, m] = velocidade_resultante.iloc[n, m].copy()
                    else:
                        contador_tipo1.iloc[posicao_vetor, m] = contador_tipo1.iloc[posicao_vetor, m] + 0
                        contador_tipo2.iloc[posicao_vetor, m] = contador_tipo2.iloc[posicao_vetor, m] + 0

print(contador_tipo1)