dataframeをリスト化し、その並び替えがうまくいかない

test.csv

csv
1testA,testB,testC
2AAA,a111,666a
3AAA,a111,555a
4BBB,a111,555a

このようなcsvファイルがあり、これを読み込みリスト化いたしました。

python
1data = pd.read_csv("test.csv")
2data =data.values

printすると下記のようになります。

python
1[['AAA' 'a111' '666a']
2 ['AAA' 'a111' '555a']
3 ['BBB' 'a111' '555a']]

これをソートすると

python
1data.sort()
2[['666a' 'AAA' 'a111']
3 ['555a' 'AAA' 'a111']
4 ['555a' 'BBB' 'a111']]

このようになってしまいます。

python
1data = [['AAA', 'a111', '666a'],
2 ['AAA', 'a111', '555a'],
3 ['BBB', 'a111', '555a']]

上記のように要素と要素の間にカンマをいれ、ソートすることで

python
1data.sort()
2[['AAA', 'a111', '555a'], ['AAA', 'a111', '666a'], ['BBB', 'a111', '555a']]

想定の並び順となりました。

dataframeをリスト化する際に、要素と要素の間にカンマをいれることは可能でしょうか？
またその方法をご教示いただけると助かります。
よろしくお願いいたします。

私の過去一個前と2個前の質問に通ずる部分なのですが、
data[np.argsort(data[:,2])]で並び替えれば良いという回答は避けていただけたら
幸いです。

melian

2021/12/17 05:22

CSV の内容とデータフレームの内容（testC列）が異なっていますが、どちらが正しいのでしょう？

jazzmaster

2021/12/17 05:26

失礼いたしました。修正いたしました。

行動規範の内容に同意します

回答1件

ベストアンサー

data.values は numpy.ndarray 型のインスタンスですので、リストが欲しい場合は tolist() メソッドで変換します。

python
1>>> import pandas as pd
2>>>
3>>> data = pd.read_csv("test.csv")
4>>> data = data.values.tolist()
5>>> data
6[['AAA', 'a111', '666a'], ['AAA', 'a111', '555a'], ['BBB', 'a111', '555a']]
7>>> type(data)
8<class 'list'>
9>>> data.sort()
10>>> data
11[['AAA', 'a111', '555a'], ['AAA', 'a111', '666a'], ['BBB', 'a111', '555a']]