Преобразование DataFrame Pandas в массив NumPy

Вы можете преобразовать DataFrame Pandas в Numpy Array для выполнения некоторых высокоуровневых математических функций, поддерживаемых пакетом Numpy.

Чтобы преобразовать DataFrame в Numpy, используйте функцию DataFrame.to_numpy(). Метод возвращает объект типа Numpy ndarray, который обычно является двумерным.

Пример 1

В следующем примере мы конвертируем DataFrame в массив numpy.

import pandas as pd

#initialize a dataframe
df = pd.DataFrame(
	[[21, 72, 67],
	[23, 78, 69],
	[32, 74, 56],
	[52, 54, 76]],
	columns=['a', 'b', 'c'])

print('DataFrame\n----------\n', df)

#convert dataframe to numpy array
arr = df.to_numpy()

print('\nNumpy Array\n----------\n', arr)

Оператор df.to_numpy() преобразует и возвращает DataFrame в массив numpy.

Пример 2: столбцы разных типов данных

Когда у вас есть DataFrame со столбцами разных типов данных, возвращаемый массив NumPy состоит из элементов одного типа данных. Самый низкий считается типом данных массива NumPy.

В следующем примере DataFrame состоит из столбцов типа int64 и float64. Когда этот DataFrame преобразуется в массив NumPy, выбирается самый низкий тип данных int64 и float64, которым является float64.

import pandas as pd
import numpy as np

#initialize a dataframe
df = pd.DataFrame(
	[[21, 72, 67.1],
	[23, 78, 69.5],
	[32, 74, 56.6],
	[52, 54, 76.2]],
	columns=['a', 'b', 'c'])

print('DataFrame\n----------\n', df)
print('\nDataFrame datatypes :\n', df.dtypes)

#convert pandas dataframe to numpy array
arr = df.to_numpy()

print('\nNumpy Array\n----------\n', arr)
print('\nNumpy Array Datatype :', arr.dtype)

Вывод:

Возвращенный массив Numpy имеет тип float64.

В этом руководстве на примерах Python мы узнали, как преобразовать Pandas DataFrame в массив Numpy.