python/numpy_array.py

45 lines
1.4 KiB
Python

import numpy as np
# Creating an array from list
array_from_list = np.array([1, 2, 3, 2, 6, 1])
# Creating an array from tuple
array_from_tuple = np.array((1, 2, 3, 2, 6, 1))
print(f'Array from list: {array_from_list}')
print(f'Array from tuple: {array_from_tuple}')
# Explicitly specifying data type
array_with_dtype = np.array([1.5, 2, 3, 2, 6, 1], dtype=float)
print(f'Array with specified dtype: {array_with_dtype.dtype.name}')
#Higher dimensional arrays
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
print(f'2D Array:\n{array_2d}')
#3d array
import numpy as np
# Creating a 3D array
array_3d = np.array([
[[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]]
])
print(f'3-dimensional array: \n{array_3d}')
#arange function
array_arange = np.arange(10) # Creates an array with values from 0 to 9
print(f'Array using arange: {array_arange}')
array_arange = np.arange(0, 30, 3) # Creates an array with values from 0 to 29 with step 3
print(f'Array using arange: {array_arange}')
# Creating an array of integers from 0 to 10 exclusive with step=1
array_1 = np.arange(11)
print(array_1)
# Creating an array of integers from 1 to 10 exclusive with step=1
array_2 = np.arange(1, 11)
print(array_2)
# Creating an array of integers from 0 to 20 exclusive with step=2
array_3 = np.arange(0, 21, 2)
print(array_3)