Q&A
ちゃんと読んでいませんでした。すみません。
対象のDataFrame
python
1import pandas as pd 2 3df = pd.DataFrame(data=[[1, 0, 1], [1, 0, 0], [1, 1, 0], [0, 1, 1]], columns=['a', 'b', 'c'])
| a | b | c |
|---|---|---|
| 1 | 0 | 1 |
| 1 | 0 | 0 |
| 1 | 1 | 0 |
| 0 | 1 | 1 |
作成したいDataFrame
抽出条件は下記の通り
１．a列b列がともに1のデータ
２．列名ではなく列番号を利用すること
| a | b | c |
|---|---|---|
| 1 | 1 | 0 |
試したこと
python
1df[df.iloc[:,[0,1]]==1].dropna(subset=df.columns[[0,1]], how='any')
| a | b | c |
|---|---|---|
| 1.0 | 1.0 | NaN |
C列についてNaNではなく元の値0を持ってくるためにはどのようにすればよいでしょうか
回答3件
0
ベストアンサー
Python
1import numpy as np 2 3print(df[np.all(df.iloc[:,[0,1]]==1, axis=1)]) 4# a b c 5#2 1 1 0
投稿2020/08/21 13:02
編集2020/08/21 13:03
総合スコア10580
0
df[(df["a"] == 1) & (df["b"] == 1)]
でどうでしょうか。
投稿2020/08/21 12:51
総合スコア369
0
列名ではなく列番号 とのことなので、iloc[:, 0] で1列目、iloc[:, 1] で2列目を取得して、(df.iloc[:, 0] == 1) & (df.iloc[:, 1] == 1) で抽出すればいいと思います。
python
1import pandas as pd 2 3df = pd.DataFrame( 4 data=[[1, 0, 1], [1, 0, 0], [1, 1, 0], [0, 1, 1]], columns=["a", "b", "c"] 5) 6df2 = df[(df.iloc[:, 0] == 1) & (df.iloc[:, 1] == 1)] 7print(df2) 8# a b c 9# 2 1 1 0
投稿2020/08/21 12:50
総合スコア21956
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。