Разделение numpy таблицы по условию

@Simply me · Регистрация: 05.05.2012

Студворк — интернет-сервис помощи студентам

Добрый день! Помогите, пожалуйста! Есть numpy-таблица с одним столбцом кортежей. Как её поделить на две, в зависимости от первого значения в кортежах, 0 или 1? Например,

Python
1
2
3
4
5
6
7
8
table = [[(0, 3, 5, 6, 7)]
         [(1, 5, 4, 3, 2)]
         [(0, 2, 6, 8, 1)]]
 
table0 = [[(0, 3, 5, 6, 7)]
          [(0, 2, 6, 8, 1)]]
 
table1 = [[(1, 5, 4, 3, 2)]]

Я бы как-то так сделала

Python
1
2
table0 = table[table[:, :][:][0]==0]
table1 = table[table[:, :][:][0]==1]

Но не работает. То есть сначала обращаюсь к кортежу таблицы, а потом к его первому элементу.

@Gdez · 25.10.2020, 18:47

Simply me, При сохранении структуры

Python
1
2
3
4
5
6
import numpy as np
table = np.array([[(0, 3, 5, 6, 7)],[(1, 5, 4, 3, 2)],[(0, 2, 6, 8, 1)]])
table1 = [[tuple(i[0].copy())] for i in table if i[0][0]]
table0 = [[tuple(i[0].copy())] for i in table if not i[0][0]]
print(table0)
print(table1)

@u235 · 26.10.2020, 05:17

Python
1
2
3
4
5
6
7
8
import numpy as np
table = np.array([[(0, 3, 5, 6, 7)],[(1, 5, 4, 3, 2)],[(0, 2, 6, 8, 1)]])
idx=table[:,0,0]==0
table0=table[idx,:,:]
idx=table[:,0,0]==1
table1=table[idx,:,:]
print(table0)
print(table1)

@Simply me · 26.10.2020, 06:53 **[ТС]**

Спасибо!

u235, в строчке

Python
1
idx=table[:,0,0]==0

пишет too many indices for array

Gdez, а можно сделать, чтобы table0 и table1 получались такого же типа, как и table? Сейчас у table тип ndarray, а у table0 и table1 тип list.

@u235 · 26.10.2020, 08:11

Simply me, так у вас похоже не кортеж, а Structured array... что выдает table.dtype.names? Как вообще задается table?

@Simply me · 26.10.2020, 08:24 **[ТС]**

Сообщение от u235

что выдает table.dtype.names?

два кортежа из названий столбцов

Сообщение от u235

Как вообще задается table?

результат вызова функции np.column_stack

Если явно преобразовать к массиву numpy, то в таблицах table0 и table1 лишние квадратные скобочки получаются
table1 = np.array([[tuple(i[0].copy())] for i in table if i[0][0]])
table0 = np.array([[tuple(i[0].copy())] for i in table if not i[0][0]])

@u235 · 26.10.2020, 08:46

Simply me, воооот, а при индексация в таких массивах нужно использовать именно эти названия, а не (не только) численные индексы. См. справку и примеры structured arrays.

@Simply me · 26.10.2020, 08:48 **[ТС]**

Спасибо!

Новые блоги и статьи Все статьи Все блоги /
Восстановить юзерскрипты Greasemonkey из бэкапа браузера damix 15.01.2026 Если восстановить из бэкапа профиль Firefox после переустановки винды, то список юзерскриптов в Greasemonkey будет пустым. Но восстановить их можно так. Для этого понадобится консольная утилита. . .	Изучаю kubernetes lagorue 13.01.2026 А пригодятся-ли мне знания kubernetes в России?	Сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/	WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .
Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .	Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .	Модель микоризы: классовый агентный подход 3 anaschu 06.01.2026 aa0a7f55b50dd51c5ec569d2d10c54f6/ O1rJuneU_ls https:/ / vkvideo. ru/ video-115721503_456239114	Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ФедосеевПавел 06.01.2026 Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ВВЕДЕНИЕ Введу сокращения: аналоговый ПИД — ПИД регулятор с управляющим выходом в виде числа в диапазоне от 0% до. . .

@u235 5515 / 2868 / 571 Регистрация: 07.11.2019 Сообщений: 4,758
	26.10.2020, 08:11
	Simply me, так у вас похоже не кортеж, а Structured array... что выдает table.dtype.names? Как вообще задается table? 0

@u235 5515 / 2868 / 571 Регистрация: 07.11.2019 Сообщений: 4,758
	26.10.2020, 08:46
	Simply me, воооот, а при индексация в таких массивах нужно использовать именно эти названия, а не (не только) численные индексы. См. справку и примеры structured arrays. 1

@Simply me 244 / 37 / 8 Регистрация: 05.05.2012 Сообщений: 521
	26.10.2020, 08:48 [ТС]
	Спасибо! 0