ValueError: cannot reshape array of size 1 into shape (2,2)

Question

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

a = pd.read_csv("test1.csv")
b= pd.read_csv("test2.csv")

data1 = a["DATA"]
data2 = b["DATA"]

data1.plot.line(title="DATA1",subplots=True, layout=(2, 2),figsize=(9, 6))
data2.plot.line(title="DATA2",subplots=True, layout=(2, 2),figsize=(9, 6))
plt.show()

とコードを書いて実行すると、上記のエラーが出ました。
グラフを2つ同時に出す構成にしたいのですが、どうしたらいいでしょうか？

Answer 1

import pandas as pd
import matplotlib.pyplot as plt

df1 = pd.DataFrame([1,2,3,4,5])
df2 = pd.DataFrame([5,4,3,2,1])

data1 = df1[0]
data2 = df2[0]

fig, axes = plt.subplots(nrows=2, ncols=1)

data1.plot.line(title="DATA1", ax=axes[0])
data2.plot.line(title="DATA2", ax=axes[1])
plt.show()

Answer 2

test1.csvとtest2.csvがどういうデータなのか分からないため、最終的なグラフのイメージ
がよく分かりませんが、DataFrameで一つにまとめて表示してはどうでしょうか？

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

a = pd.read_csv("test1.csv")
b = pd.read_csv("test2.csv")

data1 = a["DATA"]
data2 = b["DATA"]

df = pd.DataFrame({
  "DATA1": data1,
  "DATA2": data2
})

df.plot.line(subplots=True, layout=(2,2), figsize=(9,6))
plt.show()