How to beginner · 3 min read

How to create numpy array

Q: How to create numpy array

Use numpy.array() to create a numpy array from a Python list or tuple. For example, import numpy as np; arr = np.array([1, 2, 3]) creates a 1D numpy array. This array can be easily converted to a PyTorch tensor if needed.

Quick answer

Use numpy.array() to create a numpy array from a Python list or tuple. For example, import numpy as np; arr = np.array([1, 2, 3]) creates a 1D numpy array. This array can be easily converted to a PyTorch tensor if needed.

PREREQUISITES

Python 3.8+
pip install numpy>=1.23
pip install torch>=2.0

Setup

Install numpy and torch if not already installed. Use pip to install the latest versions.

bash

pip install numpy torch

Step by step

Create a numpy array from a Python list and print it. Then convert it to a PyTorch tensor to show interoperability.

python

import numpy as np
import torch

# Create a numpy array from a list
np_array = np.array([1, 2, 3, 4, 5])
print("Numpy array:", np_array)

# Convert numpy array to PyTorch tensor
torch_tensor = torch.from_numpy(np_array)
print("PyTorch tensor:", torch_tensor)

output

Numpy array: [1 2 3 4 5]
PyTorch tensor: tensor([1, 2, 3, 4, 5])

Common variations

You can create numpy arrays of different shapes and data types. Use np.zeros(), np.ones(), or np.arange() for common patterns.

python

import numpy as np

# 2D array of zeros
zeros_array = np.zeros((2, 3))
print("Zeros array:\n", zeros_array)

# 1D array of ones with float type
ones_array = np.ones(4, dtype=float)
print("Ones array:", ones_array)

# Array with range of values
range_array = np.arange(10)
print("Range array:", range_array)

output

Zeros array:
 [[0. 0. 0.]
 [0. 0. 0.]]
Ones array: [1. 1. 1. 1.]
Range array: [0 1 2 3 4 5 6 7 8 9]

Troubleshooting

If you get an error like ModuleNotFoundError: No module named 'numpy', ensure numpy is installed with pip install numpy. For type mismatch errors when converting to PyTorch, check the numpy array's dtype.

✅

Key Takeaways

Use np.array() to create numpy arrays from Python lists or tuples.
Numpy arrays can be converted to PyTorch tensors with torch.from_numpy() for seamless integration.
Common numpy array creation functions include np.zeros(), np.ones(), and np.arange().
Always verify numpy and torch installations to avoid import errors.
Check data types when converting numpy arrays to PyTorch tensors to prevent runtime errors.

Verified 2026-04

Verify ↗