Как перебирать только определенные строки в DataFrame, используя Pandas - PullRequest
2 голосов
/ 02 августа 2020

Я очень новичок в кодировании и Python, поэтому прошу прощения, если я не описываю свою проблему с использованием правильной Python терминологии.

Я создал DataFrames для 'Case' и 'X_Value' ниже.

Теперь я пытаюсь перебирать «Case» и записывать каждое «X_Value» в текстовый файл для каждого указанного c Case. Например, я хочу записать все значения X_Values ​​для случая 1 в файл (0.5, 1.2, 5.6) и все X_Values ​​в файл для варианта 2 (6.3, 9.5, 11.3). Пожалуйста помоги! Я пробовал iterrows, grouping, et c. и ничего не дает мне того результата, который я ищу. Заранее благодарим за помощь!

Case     X_Value  
1        0.5  
1        1.2  
1        5.6  
2        6.3  
2        9.5  
2        11.3  

1 Ответ

0 голосов
/ 02 августа 2020

Вы пытаетесь достичь чего-то вроде ниже?

from pandas import DataFrame


if __name__ == '__main__':
    data = {'1': [0.5, 1.2, 5.6],
            '2': [6.3, 9.5, 11.3]}
    df = DataFrame(data)

    print(df)

    for i in data.keys():
        file = "Case{}.txt".format(i)
        case = open(file, mode="w+")
        case.writelines(str(df[str(i)]))
        case.close()

Результат:

     1     2
0  0.5   6.3
1  1.2   9.5
2  5.6  11.3

Case1.txt:

0    0.5
1    1.2
2    5.6

Case2.txt

0     6.3
1     9.5
2    11.3
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...