Why does `X_train[indices.astype(int)]` give a `TypeError`?

General Discussions
,
#1

It looks like you’re getting a TypeError when trying to use the indices array in X_train[indices.astype(int)]. This error can occur if indices is not a 1D array of integers as expected.

One thing to check is the shape of indices using indices.shape. It should be (50000,), indicating a 1D array with 50000 elements.

If the shape is correct, you can try converting indices to a list before indexing:

out_images = [X_train[i] for i in indices.astype(int)]

This list comprehension should extract the elements from X_train corresponding to the indices in indices.

#2

Hey Archana,

Interesting doubt, here is the Answer to the Question:-

The error message indicates that X_train is a list, not a NumPy array, which doesn’t support array indexing. To fix this, convert X_train to a NumPy array before indexing:

out_images = np.array(X_train)[indices.astype(int)]

This converts X_train to a NumPy array and then indexes it using the integer indices from indices.astype(int).

#3

Hey Archna,

Use list comprehension to select elements from X_train based on the indices:

This iterates over the indices array and selects the corresponding elements from X_train, creating a new list out_images.

out_images = [X_train[i] for i in indices.astype(int)]

#4

Hey Archana,

Convert X_train to a NumPy array before indexing:

This first converts X_train to a NumPy array X_train_np, then indexes it using the integer indices from indices.astype(int) to get out_images.


X_train_np = np.array(X_train)
out_images = X_train_np[indices.astype(int)]