Das deutsche Python-Forum

Code: Alles auswählen

def Koordinaten(x, y, z, r, s):
    a = (100, 100, 100)
    koord = np.zeros(a)

    phi = 0
    eta = 0
    while phi <= 2pi:
        while eta <= pi:
            x = r * sin eta * cos phi
            y = r * sin eta * sin phi
            z = t * cos eta
            a[x][y][z] = s
            eta + = 0.1
    phi += 0.1

Code: Alles auswählen

import numpy as np
import math


def create_index(max_radius):
    array = np.zeros((50, 50, 50))
    index_matrix = []

    for radius in range(max_radius):
        phi = 0.0
        while phi <= math.pi * 2:
            eta = 0.0
            while eta <= math.pi:
                x = round(25 + radius * math.sin(eta) * math.cos(phi))
                y = round(25 + radius * math.sin(eta) * math.sin(phi))
                z = round(25 + radius * math.cos(eta))
                array[x,y,z] = 1.
                eta += 0.01
            phi += 0.01
        where = np.where(array == 1.)

        for i in range(3):
            arr = where[i]
            arr -= 25

        index_matrix.append(where)
        
    return index_matrix


def adjust_center(matrix, center):
    x_coord = matrix[0] + center[0]
    y_coord = matrix[1] + center[1]
    z_coord = matrix[2] + center[2]
    return tuple([x_coord, y_coord, z_coord])

index = create_index(5)
matrix = np.zeros((9,9,9))

x = 4
y = 4
z = 4
radius = 3
density = 10

idx = adjust_center(index[radius], (x,y,z))
matrix[idx] += density

for i in range(9):
    for j in range(9):
        print(matrix[i,j])

Code: Alles auswählen

a = numpy.zeros((100, 100, 100))
sz, sy, sx = (numpy.arange(s) for s in a.shape)

x, y, z, r, s = 10, 10, 10, 3, 77
mask = (sx[None,None,:]-x)**2 + (sy[None,:,None]-y)**2 + (sz[:, None, None]-z)**2 <= r
a[mask] += s