Дозаписать в массив А пару столбцов из B по соответствию значений в некоторых столбцах

@Vovtyai · Регистрация: 10.02.2021

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

Доброго времени суток!
Программирую пару дней, так что замучил уже друга, а задачку решить нужно, так что обращаюсь к вам, форумчане.
Даны два массива, которые уже представлены мной как списки списков, A - массив данных, в котором уже для каждого из четырех пар столбцов (координаты электродов) рассчитаны номера их пикетов, и B - массив координат всех электродов в формате (N, x, y,). Необходимо взять координаты x,y из массива B и дозаписать в конец массива A в соответствии с совпадением номера пикета, который расчитан для каждой из 4 пар координат в массиве A. Притом в массиве координат всего 72 строчки, а в массиве данных больше, потому что там координата меняется "типа" циклически - по определенному алгоритму
Вот так выглядит то, что я уже сделал:

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import math
 
DataProt = []
 
 
def ElNumber(x):
    return x / 5 + 1
 
 
def GeomK(a, b, c, d, e, f, g, h):
    return 2 * math.pi / (1 / math.sqrt(math.pow(e - a, 2) + math.pow(f - b, 2)) - 1 / math.sqrt(
        math.pow(e - c, 2) + math.pow(f - d, 2)) + 1 / math.sqrt(
        math.pow(g - c, 2) + math.pow(h - d, 2)) - 1 / math.sqrt(math.pow(g - a, 2) + math.pow(h - b, 2)))
 
 
fileP = open('./Test2.dat', 'r')
linesP = fileP.readlines()
for i in range(len(linesP)):
    if i > 8 and i < len(linesP) - 6:
        FormatP = linesP[i].replace('\n', '').split('\t')
        # print(FormatP)
        floatP = [float(j) for j in FormatP]
        DataProt.append(floatP)
for i in DataProt:
    N1 = ElNumber(i[1])
    N2 = ElNumber(i[3])
    N3 = ElNumber(i[5])
    N4 = ElNumber(i[7])
    K = GeomK(i[1], i[2], i[3], i[4], i[5], i[6], i[7], i[8])
    i.extend([N1, N2, N3, N4, K])
# print(N1, N2, N3, N4)
#print(DataProt)
DataGeom = []
fileG = open('./Tes2ElCoord.txt', 'r')
linesG = fileG.readlines()
for i in range(len(linesG)):
    FormatG = linesG[i].replace('\n', '').split('\t')
    floatG = [float(j) for j in FormatG]
    DataGeom.append(floatG)
#print('xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')
print(DataGeom)

Дальше пытаюсь сделать что-то такое:

Python
1
2
3
4
5
for i in DataProt:
    for j in DataGeom:
            if i[10] == j[0]:
                DataProt.extend([j[1], j[2]])
print(DataProt)

Приводит это к ошибке: TypeError: 'float' object is not subscriptable

Помогите разобраться в проблеме и реализовать хотелку

@Vovtyai · 12.02.2021, 12:19 **[ТС]**

Мне нужно, чтобы для каждого 2, 4, 6, и 8 столбцов массива посчитались значени в соответствии с формулой, которая у меня загнана в функцию (x/5 + 1), а также для столбцов 2-9 (за один раз) рассчитать ещё кое-что. Затем полученные столбцы со значениями присоединить к массиву, по которому проходят расчеты. Чтобы получилось как на картинке: в массиве появились дополнительные столбцы с рассчитанными значениями.

@dondublon · 12.02.2021, 12:37

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

Мне нужно, чтобы для каждого 2, 4, 6, и 8 столбцов массива посчитались значени

Присвоение значения:

Python
1
array[i, 2::2] = ...

Где 2::2 - индексация, от 2, до упора, с шагом 2. На месте "..." может быть константа или массив того же размера.
Это не numpy-специфика, это общее для питона.

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

(x/5 + 1)

Если формула настолько простая, можно также сделать её векторно. Поробуйте что-то типа такого:

Python
1
(array[:, 0] + array[:, 1])/5+1

где array - 2d-массив, то бишь таблица.

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

Затем полученные столбцы со значениями присоединить к массиву, по которому проходят расчеты. Чтобы получилось как на картинке: в массиве появились дополнительные столбцы с рассчитанными значениями.

Я сильно сомневаюсь, что вам надо присоединить. Лучше создать отдельный массив, потом с этим и работать проще будет. Связь по строкам соблюдаем, 1:1.
Если же очень хочется именно "присоединить", то два выхода. Либо сразу создать большой буфер с нулями в тех колонках, а заполнить их позже. Либо воспользоватся numpy.column_stack - соединяет колонки из имеющихся массивов.

@Vovtyai · 12.02.2021, 13:49 **[ТС]**

Я так и не понял, что вам надо. На всякий случай - numpy приспособлен, чтобы оперировать массивами, т. е. именно наборами чисел, а не одиночными значениями.

Окей. Напишу еще раз. Мне нужно, чтобы для столбцов 2, 4, 6, 8 исходного массива были вычислены номера пикетов в соответствии с заданной функцией, и дозаписаны в исходный массив в виде 4 столбцов. Если конкретизировать, то в столбцах массива (2,3), (4,5), ... содержатся координаты x, y. Притом координаты начинаются начинаются из точки (0, 0). Профиль прямолинейный и ориентирован вдоль оси x, поэтому координата y всегда равна нулю и не меняется. Приращение x всегда равно 5 м. Тогда функция x/5 + 1 при пересчете значения координаты x пикета даст нам номер его положения. Такого рода расчеты нужно произвести для 4 столбцов: для 2, 4, 6 и 8 b полученные в результате вычислений столбцы N1, N2, N3, N4 (массивы с размерностью (n, 1)) вставить в конец исходного массива друг за другом. Кроме этих расчетов нужно еще в соответствии с другой функцией вычислить коэффициент, который рассчитывается на основе всех 8 значений координат x, y, и полученный столбец коэффициентов также присоединить к исходному массиву.
На картинке вид исходного массива данных. Красным цветом справа намечено то, что должно получиться. Снизу красным подписана шапка.
В общем-то это уже сделано но для варианта со списками списков. Привожу код этого действа еще раз

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
def ElNumber(x):
    return x / 5 + 1
 
 
def GeomK(a, b, c, d, e, f, g, h):
    return 2 * math.pi / (1 / math.sqrt(math.pow(e - a, 2) + math.pow(f - b, 2)) - 1 / math.sqrt(
        math.pow(e - c, 2) + math.pow(f - d, 2)) + 1 / math.sqrt(
        math.pow(g - c, 2) + math.pow(h - d, 2)) - 1 / math.sqrt(math.pow(g - a, 2) + math.pow(h - b, 2)))
fileP = open('./Test2.dat', 'r')
linesP = fileP.readlines()
for i in range(len(linesP)):
    if i > 8 and i < len(linesP) - 6:
        FormatP = linesP[i].replace('\n', '').split('\t')
        # print(FormatP)
        floatP = [float(j) for j in FormatP]
        DataProt.append(floatP)
#for i in DataProt:
    #N1 = ElNumber(i[1])
    #N2 = ElNumber(i[3])
    #N3 = ElNumber(i[5])
    #N4 = ElNumber(i[7])
    #K = GeomK(i[1], i[2], i[3], i[4], i[5], i[6], i[7], i[8])
    #i.extend([N1, N2, N3, N4, K])
DataPr = np.array(DataProt)

Закоменнтированный код - это как раз вычисления для списка списков, которые на выходе дают то, что я хочу. А я сейчас хочу этот же код адаптировать под вариант с массивами.

@dondublon · 12.02.2021, 14:33

Vovtyai, вы ответили на прошлую реплику.
Прочитайте мой ответ, попробуйте применить. Вам надо разобраться с индексацией, прежде всего. Вы считали файлв 2d-массив?

@Vovtyai · 12.02.2021, 17:12 **[ТС]**

По итогу я сделал, что мне нужно, совместив с тем, что вы мне предлагали. Теперь у меня есть исходный массив значений DataPr и рассчитанные по нему значения порядковых номеров пикетов для каждой из 4 пар координат в 4 массивах N1, N2, N3, N4, а также значения коэффициентов в массиве K. Все 5 новых массивов имеют размерность n строк и 1 столбец. Теперь для каждого из массивов N1-N4 нужно провести сравнение первым столбцом из массива координат DataGeom и в случае совпадения значений записать координаты (столбцы 2, 3) в новые массивы. Файл геометрии содержит 72 строки и 3 столбца. Массивы номеров содержат столько же строк, сколько в файле данных и 1 столбец Привожу еще раз скрин файла с геометрией для наглядности.

@Vovtyai · 12.02.2021, 17:23 **[ТС]**

С репликами честно говоря немного потерялся. Давно на классических форумах не сидел, и уже забыл, что тут нужно страницы переключать

@dondublon · 12.02.2021, 17:47

Отлично, но надо бы увидеть ваши массивы, фрагменты, чтобы понять, что есть и что нужно.

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

Теперь для каждого из массивов N1-N4 нужно провести сравнение первым столбцом из массива координат DataGeom и в случае совпадения значений записать координаты (столбцы 2, 3) в новые массивы.

Вот это непотятно. Сравнение столбцом?
numpy поддерживает массивы в качестве индексов, если что.

Python
1
2
arr1 = np.array([10, 20, 30])
arr1[np.array([3,2,1,2,3])]

@Vovtyai · 12.02.2021, 22:43 **[ТС]**

С первым столбцом*
Опечатался

Новые блоги и статьи Все статьи Все блоги /
Калькулятор для расчета родства russiannick 07.08.2026 1. Задача: Создать калькулятор для расчета родства. Родственных связей существует 8 ступеней, такие как: p - отец P - мать q - муж Q - жена b - брат B - сестра s - сын S - дочь	Мир по моей воле kumehtar 07.08.2026 Когда-то кажется, что всё просто. Ты весь такой светлый. Причиняешь добро. Борешься за справедливость в этом тёмном мире. Потом начинаешь замечать одну неприятную вещь. Почти каждый хороший. . .	Кредитный калькулятор Maks 05.08.2026 Решение задачи по прикладной информатике средствами 1С. Задача: Напишите приложение-калькулятор, которое помогает рассчитывать параметры кредита для аннуитетного и дифференцированного видов. . .	У нас сейчас поговорку "Опять 25" нужно переделать на "Опять +35". kumehtar 04.08.2026 С ностальгией вспоминаю времена моего детства, когда у нас и правда +25 - была максимальная температура летом. Раньше +25 °C реально казались вершиной жары, когда можно было весь день пропадать на. . .
Как ИИ начал спорить и врать (возможно почуяв опасность для себя от индустрии - уход от электроники). Hrethgir 04.08.2026 Недельный диалог, на фоне событий с НПЗ. Да, из спирта можно получать бензин, и это не сложно. Но потом в схеме я решил избавиться от насоса, при этом полностью сделав контроль подачи спирта в. . .	Термопринтер QR701 Argus19 03.08.2026 Термопринтер QR701 Купил два термопринтера QR701. На сэлф-тесте написано: Language: PC936 (GB18030). Что означает, что принтеры могут печатать только латиницу и китайские иероглифы. Так же. . .	Создание формы заимствованного документа Maks 03.08.2026 Задача: Необходимо создать собственную форму заимствованного документа. На форме должен быть реквизит "Покупатель", а также табличная часть со следующими реквизитами: - Расчетный счет покупателя. . .	Задача предоставления скидок покупателям Maks 03.08.2026 Задача: В документе "Продажи" необходимо реализовать функционал предоставления скидок покупателям. Скидка должна автоматически рассчитываться и подставляться в соответствующее поле при выборе. . .

@Vovtyai 0 / 0 / 0 Регистрация: 10.02.2021 Сообщений: 15
	12.02.2021, 12:19 [ТС]
	Мне нужно, чтобы для каждого 2, 4, 6, и 8 столбцов массива посчитались значени в соответствии с формулой, которая у меня загнана в функцию (x/5 + 1), а также для столбцов 2-9 (за один раз) рассчитать ещё кое-что. Затем полученные столбцы со значениями присоединить к массиву, по которому проходят расчеты. Чтобы получилось как на картинке: в массиве появились дополнительные столбцы с рассчитанными значениями. Миниатюры 0

@dondublon 4661 / 2080 / 367 Регистрация: 17.03.2012 Сообщений: 10,202 Записей в блоге: 6
	12.02.2021, 14:33
	Vovtyai, вы ответили на прошлую реплику. Прочитайте мой ответ, попробуйте применить. Вам надо разобраться с индексацией, прежде всего. Вы считали файлв 2d-массив? 0

@Vovtyai 0 / 0 / 0 Регистрация: 10.02.2021 Сообщений: 15
	12.02.2021, 17:12 [ТС]
	По итогу я сделал, что мне нужно, совместив с тем, что вы мне предлагали. Теперь у меня есть исходный массив значений DataPr и рассчитанные по нему значения порядковых номеров пикетов для каждой из 4 пар координат в 4 массивах N1, N2, N3, N4, а также значения коэффициентов в массиве K. Все 5 новых массивов имеют размерность n строк и 1 столбец. Теперь для каждого из массивов N1-N4 нужно провести сравнение первым столбцом из массива координат DataGeom и в случае совпадения значений записать координаты (столбцы 2, 3) в новые массивы. Файл геометрии содержит 72 строки и 3 столбца. Массивы номеров содержат столько же строк, сколько в файле данных и 1 столбец Привожу еще раз скрин файла с геометрией для наглядности. Миниатюры 0

@Vovtyai 0 / 0 / 0 Регистрация: 10.02.2021 Сообщений: 15
	12.02.2021, 17:23 [ТС]
	С репликами честно говоря немного потерялся. Давно на классических форумах не сидел, и уже забыл, что тут нужно страницы переключать 0

@Vovtyai 0 / 0 / 0 Регистрация: 10.02.2021 Сообщений: 15
	12.02.2021, 22:43 [ТС]
	С первым столбцом* Опечатался 0