2次元配列で特定の行だけを抽出したい

Question

Pythonで2次元配列の特定の行だけを抽出したい

Python初心者です。
あるdata
data = [[1,2,3,4],[2, 2.1, 3.1, 4.1], [3, 2.2, 3.9, 5.7]...[10, 2.9, 3.9, 4.9]]
のような2次元配列があるとします。
ここで，各列に関してid, mass, temparature, densityのような値が入っています。

このとき(mass>3.4 & density>5.0)を満たす行だけを抽出したいです。

data1 = data(where((data[:, 9]< 500)&(data[:, 10] > 100)), ,delete)
のような感じで書けないかなと思いましたが上手くいきません。
そして，調べるとpandasは出てきますがそちらはデータフレームなので今回の場合に適応できないのかなと思っております。

よろしくお願いいたします。

Comments

[meg_]

質問のタイトルには「2次元配列」とありますが、質問の文章（data = [[1,2,3,4],[2, 2.1, 3.1, 4.1], [3, 2.2, 3.9, 5.7]...[10, 2.9, 3.9, 4.9]]）を見るとリストのリストです。実際のデータの型は何でしょうか？

[Lily1007]

リストのリストが2次元配列ではないのでしょうか？
またdata内の値の型はfloatを想定しております。

[Lily1007]

すみません、[[1,2,3,4][2, 2.1, 3.1, 4.1] [3, 2.2, 3.9, 5.7]...[10, 2.9, 3.9, 4.9]]
の間違いでした。(各行の表記間のコンマなし)

Answer 1

そして，調べるとpandasは出てきますがそちらはデータフレームなので今回の場合に適応できないのかなと思っております。

何故そう思われたのかは分かりませんが、下記の様な処理ならPandasで可能です。

import pandas as pd
import numpy as np

df = pd.DataFrame((np.random.randn(120)*10).reshape(30, 4), columns=['id', 'mass', 'temparature', 'density'])
print(df.head())
#	id	mass	temparature	density
#0	-3.548244	-0.497186	21.893530	-3.547576
#1	1.167749	14.301605	0.566189	-5.787389
#2	-15.907664	11.322957	5.183636	-6.337448
#3	19.463805	-17.338549	-8.904011	-13.990634
#4	-21.574818	-9.332240	-7.525486	-14.383042

result = df[(df['mass'] > 3.4) & (df['density'] > 5.0)]
print(result)
#	id	mass	temparature	density
#5	1.686821	9.970387	-3.976955	18.458043
#8	8.704518	8.367704	12.832061	12.756502
#18	7.910845	17.156489	-7.584692	15.766605
#26	0.097142	12.139886	1.813425	19.013383
#29	-0.556932	9.897616	28.229758	9.81145

Answer 2

このとき(mass>3.4 & density>5.0)を満たす行だけを抽出したいです。

data1 = data(where((data[:, 9]< 500)&(data[:, 10] > 100)), ,delete)

なぜ、3.4、5.0 が 500 や 100 になるのか分かりませんが、

(mass>3.4 & density>5.0)を満たす行だけを抽出したいのなら、
次のコードでよいのではありませんか？

data = [
	[ 1, 2,   3,   4  ],
	[ 2, 2.1, 3.1, 4.1],
	[ 3, 2.2, 3.9, 5.7],
	[ 5, 3.5, 2.0, 5.3],
	[10, 2.9, 3.9, 4.9],
	[15, 4.8, 5.0, 7.3],
];

data1 = [e for e in data if e[1]>3.4 and e[3]>5.0]
print(data1)

Comments

[Lily1007]

ありがとうございます！
> data1 = data(where((data[:, 9]< 500)&(data[:, 10] > 100)), ,delete)についてですが，500や100は例では元々応用しようとしていたdataの値をそのまま書いてしまいました。すみません。

[Lily1007]

data = [
[ 1, 2, 3, 4 ]
[ 2, 2.1, 3.1, 4.1]
[ 3, 2.2, 3.9, 5.7]
[ 5, 3.5, 2.0, 5.3]
[10, 2.9, 3.9, 4.9]
[15, 4.8, 5.0, 7.3]
]

data1 = [e for e in data if e[1]>3.4 and e[3]>5.0]
print(data1)とリストのコンマを消すとTypeError: list indices must be integers or slices, not tupleとなりますが，コンマありではないと駄目なのでしょうか。

[ppaul]

kazuma-sさんの書いたdataは、meg_さんが書かれているとおりリストのリストです。
pythonには二次元配列という概念はないのですが、多くの人が自分勝手に解釈して二次元配列と呼びます。
以下を実行してみれば、二次元配列ではないことがわかります。
コンマを消すと、エラーになる原因も少し分かるかもしれません。

data = [
[ 1, 2, 3, 4 ],
[ 2, 2.1, 3.1, 4.1],
[ 3, 2.2, 3.9, 5.7],
[ 5, 3.5, 2.0, 5.3],
[10, 2.9, 3.9, 4.9],
[15, 4.8, 5.0, 7.3],
];
data[0] = 'abc'
print(data)

[Lily1007]

listのnumpyのarrayを混同しておりました。
そもそもlistとarrayの概念の違いが分かっておらず、混乱させるような質問をしてしまい、申し訳ございません。