Skip to content
Tutorial Reference

Python NumPy: How to Convert an Image to a NumPy Array and Save It to CSV in Python

Learn how to convert an image to a NumPy array using PIL, Matplotlib, or OpenCV, then save pixel data to CSV for analysis while handling grayscale vs. RGB dimensions.

Converting images to NumPy arrays is a fundamental operation in image processing, computer vision, and machine learning. Once an image is represented as a NumPy array, you can manipulate its pixel values, apply transformations, or save the data to CSV format for storage, analysis, or sharing.

In this guide, you'll learn how to read an image into a NumPy array using different libraries, understand image array shapes, and save/load image data to and from CSV files.

Understanding Image Array Dimensions

Before diving into the conversion, it's important to understand how images are represented as arrays:

  • Grayscale images produce a 2D array with shape (height, width), where each value (0–255) represents pixel brightness.
  • Color (RGB) images produce a 3D array with shape (height, width, 3), where the three channels represent Red, Green, and Blue intensities.
Grayscale: (height, width)     → e.g., (251, 335)
RGB:       (height, width, 3)  → e.g., (251, 335, 3)

Reading an Image into a NumPy Array

Using PIL (Pillow) and NumPy

from PIL import Image
import numpy as np

img = Image.open('photo.jpg')

img_array = np.asarray(img)

print(f"Shape: {img_array.shape}")
print(f"Dtype: {img_array.dtype}")
print(f"Min: {img_array.min()}, Max: {img_array.max()}")

Output:

Shape: (251, 335, 3)
Dtype: uint8
Min: 0, Max: 255

Using Matplotlib

import matplotlib.image as mpimg

img_array = mpimg.imread('photo.jpg')

print(f"Shape: {img_array.shape}")
print(f"Dtype: {img_array.dtype}")

Output:

Shape: (251, 335, 3)
Dtype: uint8

Using OpenCV

import cv2

img_array = cv2.imread('photo.jpg')

print(f"Shape: {img_array.shape}")
print(f"Dtype: {img_array.dtype}")

img_rgb = cv2.cvtColor(img_array, cv2.COLOR_BGR2RGB)

Output:

Shape: (251, 335, 3)
Dtype: uint8

Saving an Image Array to CSV

CSV files can only store 2D data (rows and columns). Since color images are 3D arrays, you need to reshape the array to 2D before saving.

Method 1: Using NumPy (savetxt / loadtxt)

import numpy as np
from PIL import Image

img = Image.open('photo.jpg')
img_array = np.asarray(img)
print(f"Original shape: {img_array.shape}")

if img_array.ndim == 3:
    img_2d = img_array.reshape(img_array.shape[0], -1)
    print(f"Reshaped to 2D: {img_2d.shape}")
else:
    img_2d = img_array  # Grayscale is already 2D

np.savetxt('image_data.csv', img_2d, delimiter=',', fmt='%d')
print("Saved to image_data.csv")

Output:

Original shape: (251, 335, 3)
Reshaped to 2D: (251, 1005)
Saved to image_data.csv

How the reshape works:

A (251, 335, 3) image becomes (251, 1005) because each row of 335 pixels × 3 channels = 1005 values. The pixel data is flattened as: [R1, G1, B1, R2, G2, B2, ..., R335, G335, B335] for each row.

Method 2: Using Pandas

import numpy as np
import pandas as pd
from PIL import Image

img = Image.open('photo.jpg')
img_array = np.asarray(img)

if img_array.ndim == 3:
    img_2d = img_array.reshape(img_array.shape[0], -1)
else:
    img_2d = img_array

df = pd.DataFrame(img_2d)
df.to_csv('image_data.csv', header=False, index=False)
print(f"Saved {img_2d.shape} array to CSV")

Output:

Saved (251, 1005) array to CSV

Loading Image Data from CSV

Using NumPy

import numpy as np

loaded_2d = np.loadtxt('image_data.csv', delimiter=',')

original_height = 251
original_width = 335
num_channels = 3

loaded_image = loaded_2d.reshape(original_height, original_width, num_channels)
loaded_image = loaded_image.astype(np.uint8)

print(f"Loaded shape: {loaded_image.shape}")
print(f"Dtype: {loaded_image.dtype}")

Output:

Loaded shape: (251, 335, 3)
Dtype: uint8

Using Pandas

import numpy as np
import pandas as pd

loaded_df = pd.read_csv('image_data.csv', header=None)
loaded_2d = loaded_df.values

original_channels = 3
original_width = loaded_2d.shape[1] // original_channels

loaded_image = loaded_2d.reshape(loaded_2d.shape[0], original_width, original_channels)
loaded_image = loaded_image.astype(np.uint8)

print(f"Loaded shape: {loaded_image.shape}")

Output:

Loaded shape: (251, 335, 3)

Complete Workflow: Save and Verify

Here's a complete, reusable workflow that saves an image to CSV and verifies the data integrity upon loading:

import numpy as np
from PIL import Image

def image_to_csv(image_path, csv_path):
    """Convert an image to a NumPy array and save to CSV."""
    img = Image.open(image_path)
    img_array = np.asarray(img)

    original_shape = img_array.shape
    print(f"Original shape: {original_shape}")

    if img_array.ndim == 3:
        img_2d = img_array.reshape(img_array.shape[0], -1)
    else:
        img_2d = img_array

    np.savetxt(csv_path, img_2d, delimiter=',', fmt='%d')
    print(f"Saved to {csv_path} (shape: {img_2d.shape})")

    return original_shape


def csv_to_image(csv_path, original_shape):
    """Load image data from CSV and reconstruct the array."""
    loaded_2d = np.loadtxt(csv_path, delimiter=',')

    loaded_image = loaded_2d.reshape(original_shape).astype(np.uint8)
    print(f"Loaded shape: {loaded_image.shape}")

    return loaded_image


shape = image_to_csv('photo.jpg', 'image_data.csv')

loaded = csv_to_image('image_data.csv', shape)

original = np.asarray(Image.open('photo.jpg'))
if np.array_equal(original, loaded):
    print("\n✅ Data integrity verified: loaded data matches original.")
else:
    print("\n❌ Data mismatch detected.")

Output:

Original shape: (251, 335, 3)
Saved to image_data.csv (shape: (251, 1005))
Loaded shape: (251, 335, 3)

✅ Data integrity verified: loaded data matches original.

Important Considerations

Conclusion

Converting images to NumPy arrays and saving them to CSV involves three key steps:

  1. Read the image using PIL, Matplotlib, or OpenCV to get a NumPy array.
  2. Reshape 3D arrays to 2D since CSV files only support two-dimensional data - flatten the width and channel dimensions into a single axis.
  3. Save with np.savetxt() or pandas to_csv(), and reconstruct the original shape when loading.

For production workflows, consider using NumPy's native .npy format instead of CSV for significantly better performance and smaller file sizes. Reserve CSV for cases where human readability or cross-platform compatibility is important.

Navigation