リストの要素値がすべて0である行と列を削除したい

Question

実現したいこと

こちらの関連質問です。

R2.csvは以下のようなファイルです。
1 2 0
3 0 0
0 0 0

R2の要素値がすべて0である行と列を削除したいです。

こちらのように、すべて0である行は削除できました。「該当のソースコード」の18行目のように、axis=1をaxis=0にすればうまくいくだろうと思っていたのですが、実際にはうまくいきませんでした。どなたかコードを直してください。

発生している問題・エラーメッセージ

〇〇〇\testCV526.py:18: UserWarning: Boolean Series key will be reindexed to match DataFrame index.
  R2 = R2[~R2.eq(0).all(axis=0)].to_numpy()

該当のソースコード

import pandas as pd

R2 = pd.read_csv("R2.csv",header=None)
print(R2)
#   0  1  2
#0  1  2  0
#1  3  0  0
#2  0  0  0

# 値が全て0の行を削除
R2 = R2[~R2.eq(0).all(axis=1)]
print(R2)
#   0  1  2
#0  1  2  0
#1  3  0  0

# 値が全て0の列を削除
R2 = R2[~R2.eq(0).all(axis=0)].to_numpy()
print(R2)
#[[1 2 0]
# [3 0 0]]

試したこと

axisの値を変えるだけでうまくいくと思っていたので、何を調べたら良いか分かりませんでした。

こちらを参考にして、18行目を
R2 = R2[:, ~R2.eq(0).all(axis=0)].to_numpy()
としてみましたが、うまくいきませんでした。

Answer 1

行のみの場合 と同様の記述例を下記に示します。なお，行と列を同時に指定するために np.ix_() を用いました。

import numpy as np

R2 = np.loadtxt('R2.csv', dtype=int)
print(R2)
# [[1 2 0]
#  [3 0 0]
#  [0 0 0]]

R2 = R2[np.ix_(~np.all(R2 == 0, axis=1), ~np.all(R2 == 0, axis=0))]
print(R2)
# [[1 2]
#  [3 0]]

Comments

[yyicp]

「行のみの場合」と同様に、私の環境では
R2 = np.loadtxt('R2.csv', dtype=int, delimiter=',')
とすることで、できました。

どうもありがとうございました。

Answer 2

loc メソッドを使います。

R2 = R2.loc[~R2.eq(0).all(axis=1), ~R2.eq(0).all(axis=0)]
print(R2)

#    0  1
# 0  1  2
# 1  3  0

Comments

[yyicp]

できました。
どうもありがとうございました。