В этой статье блога мы рассмотрим различные методы добавления 2D-массива в DataFrame в Python. DataFrame — это популярная структура данных, предоставляемая библиотекой pandas, и возможность добавления к ней двумерного массива может быть полезна во многих задачах манипулирования данными. Мы обсудим пять различных подходов вместе с примерами кода, которые помогут вам понять и реализовать их в своих проектах.

Методы:

1. Использование конструктора DataFrame:
   Вы можете создать DataFrame непосредственно из двумерного массива, передав массив в качестве аргумента конструктору DataFrame. Вот пример:

import pandas as pd
import numpy as np
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame(array_2d)

1. Преобразование 2D-массива в список словарей:
   Вы можете преобразовать каждую строку 2D-массива в словарь, а затем создать DataFrame из списка словарей. Вот пример:

import pandas as pd
import numpy as np
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
dict_list = [{f'col{i}': row[i] for i in range(len(row))} for row in array_2d]
df = pd.DataFrame(dict_list)

1. Использование метода Append():
   Вы можете создать пустой DataFrame, а затем добавить строки из 2D-массива с помощью метода Append(). Вот пример:

import pandas as pd
import numpy as np
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame()
for row in array_2d:
    df = df.append(pd.Series(row), ignore_index=True)

1. Использование метода from_records():
   Вы можете использовать метод from_records(), предоставляемый библиотекой pandas, для создания DataFrame непосредственно из 2D-массива. Вот пример:

import pandas as pd
import numpy as np
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame.from_records(array_2d)

1. Непосредственное использование numpy ndarray:
   Если вы уже работаете с numpy ndarray, вы можете преобразовать его в DataFrame без явного использования библиотеки pandas. Вот пример:

import pandas as pd
import numpy as np
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame(array_2d.tolist())

В этой статье мы рассмотрели пять различных методов добавления двумерного массива в DataFrame в Python. Используя возможности панд и библиотек numpy, вы можете выбрать метод, который соответствует вашему конкретному случаю использования. Предпочитаете ли вы напрямую использовать конструктор DataFrame, преобразовывать массив в список словарей или использовать универсальные функции, доступные в pandas, теперь у вас есть подробное руководство для эффективного выполнения этой операции.