Добавить несколько столбцов из df в список

Question

Я пытаюсь import xy coordinates из нескольких columns в список.Я могу начать, когда coordinates читается из одного column, но изо всех сил пытается заставить его читать из нескольких columns эффективно.

Мне нужно это для построения

Попытка:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

d = ({
    'Time' : [1,2,3,4,5,6,7,8],       
    'GrA1_X' : [10,12,17,16,16,14,12,8],                 
    'GrA1_Y' : [10,12,13,7,6,7,8,8], 
    'GrA2_X' : [5,8,13,16,19,15,13,5],                 
    'GrA2_Y' : [6,15,12,7,8,9,10,8],
    'GrB1_X' : [15,18,25,16,6,15,17,10],                 
    'GrB1_Y' : [7,12,5,9,10,12,18,9], 
    'GrB2_X' : [10,4,18,14,16,12,17,4],                 
    'GrB2_Y' : [8,12,16,8,10,14,12,15],         
     })

df = pd.DataFrame(data=d)

GrA_X = df[df.columns[1::2][:2]]
GrA_Y = df[df.columns[2::2][:2]]

GrB_X = df[df.columns[5::2][:2]]
GrB_Y = df[df.columns[6::2][:2]]

fig = plt.figure(figsize=(10,6))
ax = plt.gca()

Zs = []
for l,color in zip('AB', ('red', 'blue')):
    # plot all of the points from a single group
    ax.plot(GrA_X.iloc[0], GrA_Y.iloc[0], '.', c='red', ms=15, label=l, alpha = 0.5)
    ax.plot(GrB_X.iloc[0], GrB_Y.iloc[0], '.', c='blue', ms=15, label=l, alpha = 0.5)    

    Zrows = []
    for _,row in df.iterrows():
        x,y = row['Gr%s_X'%l], row['Gr%s_Y'%l]

Я застрял с вызовом Zrows = [].В частности, как добавить multiple columns в этот список.

Answer 1

Это может быть альтернативным решением, если я правильно понимаю вашу проблему.

df = pd.DataFrame(data=d)
X = [df[c].tolist() for c in df.columns if c.find("_X") >= 0]
Y = [df[c].tolist() for c in df.columns if c.find("_Y") >= 0]

allX = [x for sublist in X for x in sublist]
allY = [y for sublist in Y for y in sublist]

dfXY = pd.DataFrame({"X": allX, "Y":allY})

Теперь у вас есть все x и y в простом кадре данных.Приветствия

Answer 2

Мне все еще не ясно, правильно ли я понимаю цель последнего цикла, но я рискну и предложу решение.Похоже, ваша проблема связана с именованием столбцов, и я не вижу другого пути, кроме циклического перебора различных вариантов Gr{A,B}{1,2}_{X,Y} (здесь {} указывает на варианты).Я бы пошел с вложенным for -циклом, то есть:

Zs = []
for l,color in zip('AB', ('red', 'blue')):
    # plot all of the points from a single group
    ax.plot(GrA_X.iloc[0], GrA_Y.iloc[0], '.', c='red', ms=15, label=l, alpha = 0.5)
    ax.plot(GrB_X.iloc[0], GrB_Y.iloc[0], '.', c='blue', ms=15, label=l, alpha = 0.5)    

    Zrows = []
    for _,row in df.iterrows():
        for i in [1,2]:
            x,y = row['Gr{}{}_X'.format(l,i)], row['Gr{}{}_Y'.format(l,i)]
            Zrows.append((x,y))


print(Zrows)

, который дает список кортежей координат:

[(15, 7), (10, 8), (18, 12), (4, 12), (25, 5), (18, 16), (16, 9), (14, 8), (6, 10), (16, 10), (15, 12), (12, 14), (17, 18), (17, 12), (10, 9), (4, 15)]

Обратите внимание, что я изменил форматирование строки на format() -синтаксис, который, на мой взгляд, значительно повышает читабельность.Пожалуйста, дайте мне знать, если это даже близко к тому, что вы после.