質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

90.12%

特定の行の要素を取得したい

解決済

回答 2

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 92

farinelli

score 14

文章にすると長くなりそうなので箇条書きにします。

・後半2列がランダムな文字列になっている6列からなるcsvファイルから2次元配列を取得する
・1~3つ目と4~5つ目を異なるリストとして扱う
・最大値がその行の2番目に大きい値の2倍を超える行に限り、「0番目の要素, 最大値, (0番目の要素と同じものを含む最後の要素の左隣に位置する)最後から2番目の要素, (0番目の要素と同じものを含む)最後の要素」の順番で取得する(ここの部分が完全でない)

この条件を満たすコードを書きたいのですが、エラーが出て困っています。
お知恵をお越しいただけますと幸いです。宜しくお願いします。

エラー

  File "comp.py", line 11, in <module>
    data = np.loadtxt("renshu2.csv", delimiter = ",", skiprows=1)
  File "/Users/macuser/.pyenv/versions/3.7.4/lib/python3.7/site-packages/numpy/lib/npyio.py", line 1146, in loadtxt
    for x in read_data(_loadtxt_chunksize):
  File "/Users/macuser/.pyenv/versions/3.7.4/lib/python3.7/site-packages/numpy/lib/npyio.py", line 1074, in read_data
    items = [conv(val) for (conv, val) in zip(converters, vals)]
  File "/Users/macuser/.pyenv/versions/3.7.4/lib/python3.7/site-packages/numpy/lib/npyio.py", line 1074, in <listcomp>
    items = [conv(val) for (conv, val) in zip(converters, vals)]
  File "/Users/macuser/.pyenv/versions/3.7.4/lib/python3.7/site-packages/numpy/lib/npyio.py", line 781, in floatconv
    return float(x)
ValueError: could not convert string to float: 'A'```  

コード

import pandas as pd  
import numpy as np  

df = np.loadtxt("renshu2.csv", delimiter = ",", skiprows=1)  
para = df[1:3]  
num = [int(s) for s in para]  
id = df[4:]  
for idx, row in num.iterrows():  
score = row.sort_values(ascending=False)  
if score.iloc[0] >= score.iloc[1] * 2:  
print([idx, score.index[0], id])  

ファイル

Name,One,Two,Three,Group name,Group member  
A,3,3,14,杉田,"B,F,R"  
B,3,1,2,中村,"A,Q,V"  
C,3,2,2,宮野,"C,J,N"  

理想の出力

[A, 14, 中村, "A,Q,V"]  

補足情報(FW/ツールのバージョンなど)

iOS 10.14.6, Python 3.7.4, Atom

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

質問への追記・修正、ベストアンサー選択の依頼

  • magichan

    2019/08/29 16:44

    「最大値がその行の2番目に大きい値の2倍を超える行に限り」とありますが、この条件を満たさない行はどのように扱うのでしょうか?(行を削除するのでしょうか)
    理想となる出力を記述してください

    キャンセル

  • farinelli

    2019/08/29 17:51

    一部修正しましたが、疑問等ございましたら再度コメントいただけますと幸いです。

    キャンセル

回答 2

checkベストアンサー

+1

これで良いかな

昨日の質問の応用です

import pandas as pd
df = pd.read_csv('renshu2.csv', index_col='Name')
for idx, row in df.iterrows():
    # score は 'One'列~'Three'列の間の値から導く
    score = row['One':'Three'].sort_values(ascending=False)
    if score.iloc[0] >= score.iloc[1] * 2:
        print([idx, score.iloc[0], row['Group name'], row['Group member']])

【仕様にあわせて修正】

import pandas as pd
df = pd.read_csv('renshu2.csv', index_col='Name')
for idx, row in df.iterrows():
    score = row['One':'Three'].sort_values(ascending=False)
    if score.iloc[0] >= score.iloc[1] * 2:
        # 'Group member'に idx を含んでる行を求める
        target_row = df[df['Group member'].str.contains(idx)].iloc[0]
        print([idx, score.iloc[0], target_row['Group name'], target_row['Group member']])

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2019/08/29 18:26

    回答有難うございます。
    後出しですみません。先程依頼文を一部修正したのですが、「理想の出力」を例に例えると、数字の面で条件を満たすのはNameがAの行であり、最終的には6番目にAを含む"A,Q,V"と"A,Q,V"と同じ行にあった5番目の「中村」という要素をNameがAの行に引っ張ってきたいのです。
    提示していただいたコードの出力結果は['A', 14, '杉田', 'B,F,R']というものであり、「理想の出力」にある[A, 14, 中村, "A,Q,V"] とは少しズレが生じてしまいます。

    キャンセル

  • 2019/08/29 19:10

    ワークしました!有難うございます!

    キャンセル

0

「np.loadtxt()」ではなく、「pd.read_csv()」を使えば良いと思います。

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2019/08/29 13:12

    有難うございます。
    そうしました所、num = [int(s) for s in para]の行でValueError: invalid literal for int() with base 10: 'One'と出てしまいました。
    1行目を無視できるようなpandasのオプション等、ご存知でしたらご教授いただけますと幸いです。

    キャンセル

  • 2019/08/29 18:13

    df[1:3]は行を抽出したDataFrameとなります。
    for s in paraで列名を操作しています。
    やりたいこととコードは一致していますか?

    キャンセル

  • 2019/08/29 19:10

    すみません、無意識に書いてました。直します。
    ご指摘有難うございます。

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 90.12%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる