import numpy as np
import time

# Python list
list_a = list(range(1000000))
list_b = list(range(1000000))
start = time.time()
result = [x + y for x, y in zip(list_a, list_b)]
print(f"List time: {time.time() - start:.3f}s")

# NumPy array
arr_a = np.arange(1000000)
arr_b = np.arange(1000000)
start = time.time()
result = arr_a + arr_b
print(f"NumPy time: {time.time() - start:.3f}s")

import sys
import numpy as np

python_list = [0] * 1000
numpy_array = np.zeros(1000, dtype=np.int32)

print(f"List size: {sys.getsizeof(python_list) + sum(sys.getsizeof(i) for i in python_list)} bytes")
print(f"NumPy size: {numpy_array.nbytes} bytes")

import numpy as np
import time

size = 10000000
arr = np.random.random(size)
lst = list(arr)

start = time.time()
np_sum = np.sum(arr)  # Truy cập liên tục
print(f"NumPy sum: {time.time() - start:.3f}s")

start = time.time()
lst_sum = sum(lst)  # Truy cập rải rác
print(f"List sum: {time.time() - start:.3f}s")

arr = np.array([1, 2, 3])
print(arr * 2)  # Output: [2 4 6]

a = np.array([1, 2, 3])
b = np.array([[1, 1, 1], [2, 2, 2]])
print(a * b)
# Output:
# [[1 2 3]
#  [2 4 6]]

arr = np.array([10, 20, 30, 40])
print(arr[arr > 25])  # Output: [30 40]