Q&A
ある多次元配列があり、特定の値のみを抽出し新しい配列を作りたいと考えています。
やりたいこと
以下のような配列から、
a b c d 1 [ -0.8, -0.4, 0.3, 8. ], 2 [ -0.1, 0.8, -0.4, 5. ], 3 [ 0.8, -0.1, 0.4 , 15. ], 4 [ 0.7, 0.5, -0.3, 3. ], 5 [ -0.6, 0.6, 0.3, 11. ], 6 [ 0.0, 0.1, -0.9, 1. ]
dの要素に8もしくは1の値を持つ行を抽出し、以下のような配列を作りたいです。
a b c d 1 [ -0.8, -0.4, 0.3, 8. ], 2 [ 0.0, 0.1, -0.9, 1. ]
if文を用いてループしようと考えておりましたが、その場合、ラベルの数だけループを回す必要があり、無駄を感じたため、何か他に良い方法がないか知りたいため、質問させていただきました。
回答2件
0
ベストアンサー
準備。
Python
1import pandas as pd 2import numpy as np 3 4df = pd.DataFrame( 5np.array([[ -0.8, -0.4, 0.3, 8. ], 6[ -0.1, 0.8, -0.4, 5. ], 7[ 0.8, -0.1, 0.4 , 15. ], 8[ 0.7, 0.5, -0.3, 3. ], 9[ -0.6, 0.6, 0.3, 11. ], 10[ 0.0, 0.1, -0.9, 1. ], 11]), columns=list('abcd'))
データフレームはboolの配列で選択できます。
Python
1>>> df[[True, False, False, False, False, True]] 2 a b c d 30 -0.8 -0.4 0.3 8.0 45 0.0 0.1 -0.9 1.0
データフレームは列名で列を取り出せます。
Python
1>>> df.d 20 8.0 31 5.0 42 15.0 53 3.0 64 11.0 75 1.0 8Name: d, dtype: float64
データフレームは比較演算子でboolの配列が取り出せます。
Python
1>>> df.d == 8.0 20 True 31 False 42 False 53 False 64 False 75 False 8Name: d, dtype: bool
boolのデータフレームは | で論理和ができます。結合の優先度の関係で(?)括弧が必要です。
Python
1>>> (df.d == 8.0) | (df.d == 1.0) 20 True 31 False 42 False 53 False 64 False 75 True 8Name: d, dtype: bool
以上から、必要な操作はこうなります。
Python
1>>> df[(df.d == 8.0) | (df.d == 1.0)] 2 a b c d 30 -0.8 -0.4 0.3 8.0 45 0.0 0.1 -0.9 1.0
投稿2018/06/15 11:58
総合スコア11051
0
numpy使えば以下のようにできます。
もっとスマートな書き方もあるはずですが（思い出せない）。
Python
1import numpy as np 2 3a = np.array([ 4 [ -0.8, -0.4, 0.3, 8.], 5 [ -0.1, 0.8, -0.4, 5.], 6 [ 0.8, -0.1, 0.4 , 15.], 7 [ 0.7, 0.5, -0.3, 3.], 8 [ -0.6, 0.6, 0.3, 11.], 9 [ 0.0, 0.1, -0.9, 1.] 10 ]) 11print(a) 12 13b = a[(a[:,3] == 1) | (a[:,3] == 8)] # 3列目が1または8 14print(b) 15""" 16[[-0.8 -0.4 0.3 8. ] 17 [ 0. 0.1 -0.9 1. ]] 18"""
投稿2018/06/15 09:12
総合スコア38266
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2018/06/15 12:53