Question

How to create and set values to a 3d array in Python

Answer 1

import numpy as np

def initialize_array(lst3d, n):
    for i in range(n):
        for j in range(n):
            for k in range(n):
                lst3d[i][j][k] = i + j + k  # Initialization

n = 3

# Create a 3x3x3 3D array initialized with zeros
array3d = np.zeros((n, n, n))

initialize_array(array3d, n)

# Set values
array3d[0, 0, 0] = 898
array3d[1, 1, 1] = 120
array3d[2, 2, 2] = 731

print(array3d)



'''
run:

[[[898.   1.   2.]
  [  1.   2.   3.]
  [  2.   3.   4.]]

 [[  1.   2.   3.]
  [  2. 120.   4.]
  [  3.   4.   5.]]

 [[  2.   3.   4.]
  [  3.   4.   5.]
  [  4.   5. 731.]]]

'''

 

Learn SQL from A to Z at LearnSQL.com