【Python × マーベル映画】pandas データフレームの作り方
こんにちは!こちらの記事では、Pandas を使ってデータフレームを作成する方法を紹介します。
環境: Python 3.9, Pandas 1.2.4
- 1. リストからDataFrameを作成する。
- 2. 複数のリストからDataFrameを作成する。
- 3. 辞書のリストからDataFrameを作成する。
- 4. インデックスをつけて、データフレームを作成します。
1. リストからDataFrameを作成する。
import pandas as pd
from IPython.display import display
#リストのリストを作成
list_of_lists = [[1, "Iron Man", "Robert Downey, Jr."],
[2, "The Incredible Hulk", "Edward Norton"],
[3, "Iron Man 2", "Robert Downey Jr."],
[4, "Thor", "Chris Hemsworth"]]
# データフレームを作成
df_a = pd.DataFrame(list_of_lists, columns=["FilmID", "FilmName","MainActor"])
display(df_a)
2. 複数のリストからDataFrameを作成する。
# Import library
import pandas as pd
from IPython.display import display
# 複数のリストを作成
Film_ID = [1, 2, 3, 4]
Film_name = ["Iron Man", "The Incredible Hulk", "Iron Man 2","Thor"]
Main_actor = ["Robert Downey Jr.", "Edward Norton", "Robert Downey Jr.", "Chris Hemsworth"]
# データフレームを作成
df_b = pd.DataFrame(list(zip(Film_ID, Film_name, Main_actor)),
columns=["FilmID", "FilmName","MainActor"])
display(df_b)
3. 辞書のリストからDataFrameを作成する。
import pandas as pd
from IPython.display import display
# 辞書のリストを作成
dictionary_of_lists = {
"FilmID": [1, 2, 3, 4],
"FilmName": ["Iron Man", "The Incredible Hulk", "Iron Man 2","Thor"],
"MainActor": ["Robert Downey Jr.", "Edward Norton", "Robert Downey Jr.", "Chris Hemsworth"]}
# データフレームを作成
df_c = pd.DataFrame(dictionary_of_lists)
display(df_c)
4. インデックスをつけて、データフレームを作成します。
import pandas as pd
from IPython.display import display
# インデックスをつけて、データフレームを作成
df_d = pd.DataFrame(columns=["FilmID", "FilmName","ActorName"], index=['a', 'b', 'c','d'])
df_d.loc['a'] = [1, "Iron Man","Robert Downey Jr."] #Rober"s name should be counted as three pieces
df_d.loc['b'] = [2, "The Incredible Hulk", "Mark Ruffalo"]
df_d.loc['c'] = [3, "Iron Man 2", "Robert Downey Jr."]
df_d.loc['d'] = [4, "Thor", "Chris Hemsworth"]
display(df_d)
今回は、以上となります。ありがとうございました。