Разделение 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 **[ТС]**

Спасибо!

Новые блоги и статьи Все статьи Все блоги /
Отчёт о затраченных материалах за определенный период с макетом печатной формы Maks 21.04.2026 Отчёт из решения ниже размещён в конфигурации КА2. Задача: показать затраченные материалы за определённый период, с возможностью вывода печатной формы отчёта с шапкой и подвалом. В качестве. . .	Отчёт о спецтехнике находящейся в ремонте Maks 20.04.2026 Отчёт из решения ниже размещен в конфигурации КА2. Задача: отобразить спецтехнику, которая на данный момент находится в ремонте. Есть нетиповой документ "Заявка на ремонт спецтехники" который. . .	Памятка для бота и "визитка" для читателей "Semantic Universe Layer (Слой семантической вселенной)" Hrethgir 19.04.2026 Сгенерировано для краткого описания по случаю сборки и компиляции скелета серверного приложения. И пусть после этого скажут, что статьи сгенерированные AI - туфта и не интересно. И это не реклама -. . .	Запрет удаления строк ТЧ документа при определённом условии Maks 19.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "Аккумуляторы", разработанного в конфигурации КА2. У данного документа есть ТЧ, в которой в зависимости от прав доступа. . .
Модель заражения группы наркоманов alhaos 17.04.2026 Условия задачи сформулированы тут Суть: - Группа наркоманов из 10 человек. - Только один инфицирован ВИЧ. - Колются одной иглой. - Колются раз в день. - Колются последовательно через. . .	Мысли в слух. Про "навсегда". kumehtar 16.04.2026 Подумалось тут, что наверное очень глупо использовать во всяких своих установках понятие "навсегда". Это очень сильное понятие, и я только начинаю понимать край его смысла, не смотря на то что давно. . .	My Business CRM MaGz GoLd 16.04.2026 Всем привет, недавно возникла потребность создать CRM, для личных нужд. Собственно программа предоставляет из себя базу данных клиентов, в которой можно фиксировать звонки, стадии сделки, а также. . .	Знаешь почему 90% людей редко бывают счастливыми? kumehtar 14.04.2026 Потому что они ждут. Ждут выходных, ждут отпуска, ждут удачного момента. . . а удачный момент так и не приходит.

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

@u235 5521 / 2874 / 572 Регистрация: 07.11.2019 Сообщений: 4,770
	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

Опции темы