Преобразуйте следующий код из Matlab в Python

Question

Я новичок в Python и пытаюсь преобразовать и изменить следующий код из Matlab в python:

Это то, что у меня есть до сих пор (я также пытаюсь сделатьэто для 3 измерений):

import random
import numpy as np

L = 21.1632573
x = np.random.uniform(low=0.0000,high=L,size=10000) 
y = np.random.uniform(low=0.0000,high=L,size=10000) 
z = np.random.uniform(low=0.0000,high=L,size=10000) 


prox = 1
N = 20

#First Point
firstX = x[0]
firstY = y[0]
firstZ = z[0]

counter = 0
for k in range(1,N):
    thisX = x[k]
    thisY = y[k]
    thisZ = z[k]
    distances = np.sqrt((thisX-firstX)**2+(thisY-firstY)**2+(thisZ-firstZ)**2)
    minDistance = np.min(distances)
    if minDistance >= prox:
        firstX[counter] = thisX
        firstY[counter] = thisY
        firstZ[counter] = thisZ
        counter = counter + 1

Однако у меня возникла проблема в последнем операторе if:

File "/home/aperego/codes/LJ_Problem1/canonical/randomParticles.py", 
line 26, in <module> firstX[counter] = thisX

TypeError: 'numpy.float64' object does not support item assignment

Любая помощь приветствуется!

Спасибо

Answer 1

вы присваиваете переменные numpy float.эти переменные должны быть в списке

firstX = x[0]  # all numpy.float64
firstY = y[0]
firstZ = z[0]

вы должны добавить новую точку в список

import random
import numpy as np

L = 21.1632573
x = np.random.uniform(low=0.0000,high=L,size=10000) 
y = np.random.uniform(low=0.0000,high=L,size=10000) 
z = np.random.uniform(low=0.0000,high=L,size=10000) 


prox = 1
N = 20

#First Point
firstX = [x[0]]
firstY = [y[0]]
firstZ = [z[0]]

for k in range(1,N):
    thisX = x[k]
    thisY = y[k]
    thisZ = z[k]
    distances = np.sqrt((thisX-firstX[0])**2+(thisY-firstY[0])**2+(thisZ-firstZ[0])**2)
    minDistance = np.min(distances)
    if minDistance >= prox:
        firstX.append(thisX)
        firstY.append(thisY)
        first.append(thisZ)

Answer 2

firstX, firstY и firstZ являются числами, поэтому вы не можете использовать firstX [index].Поэтому определите их как список или массив (если вы знаете конечную длину).

Я прочитал ваш код Matlab, исправил его и вычеркнул соответственно.

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

L = 21.1632573
x = np.random.uniform(low=0.0000,high=L,size=10000) 
y = np.random.uniform(low=0.0000,high=L,size=10000) 
z = np.random.uniform(low=0.0000,high=L,size=10000) 

prox = 1
N = 20

#First Point
firstX = [x[0]]
firstY = [y[0]]
firstZ = [z[0]]

counter = 0
for k in range(1,N):
    distances = np.sqrt((x[k]-firstX)**2+(y[k]-firstY)**2+(z[k]-firstZ)**2)
    minDistance = np.min(distances)
    if minDistance >= prox:
        firstX.append(x[k])
        firstY.append(y[k])
        firstZ.append(z[k])
        counter = counter + 1

##Plotting
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.scatter(firstX,firstY,firstZ, c='b', marker='*')
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')

plt.show()