Гость
Войти | Профиль | Очистить
Целевая тема:
Создать новую тему:
Автор:
Действия ...
Форумы / PHP, Perl, Python [игнор отключен] [закрыт для гостей] / Python. Замена значений в np.array / 3 сообщений из 3, страница 1 из 1
05.02.2021, 19:54
    #40042337
ferzmikk
ferzmikk
Участник
Скрыть профиль Поместить в игнор-лист Сообщения автора в теме
Python. Замена значений в np.array
Здравствуйте!

Есть заданный массив:
Код: python
1.
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]


и словарь для замены:
Код: python
1.
{10 : 6, 6 : 7}


В массиве нужно изменить значения на основе словаря. Результат должен быть таким:
Код: python
1.
[1 2 3 4 5 7 7 8 9 6]


Если так написать:
Код: python
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
import numpy as np

array = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

dict = {10 : 6, 6 : 7}

for element in dict.keys():    
    np.place(array, array == element, dict.get(element))

print(array)

Выдает такой результат
Код: python
1.
[1 2 3 4 5 7 7 8 9 7]

Это неправильно.


Пишу по другому. С использованием Pandas.
Код: python
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
import numpy as np
import pandas as pd

array = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

dict = {10 : 6, 6 : 7}

array = np.expand_dims(array, axis = 1) 

df = pd.DataFrame(data=array)
df[0].replace(dict, inplace=True)
array = df.values
array = np.squeeze(array, axis = 1)

print(array)

Выдает такой результат
Код: python
1.
[1 2 3 4 5 7 7 8 9 6]

Это правильно.

Второй вариант получился не оптимальным, то есть перегруженный код. Возможно ли написать короче с использованием только Numpy?
...
Рейтинг: 0 / 0
| Ответить | Цитировать | Написать  
07.02.2021, 05:12
    #40042755
vikkiv
vikkiv
Участник
Скрыть профиль Поместить в игнор-лист Сообщения автора в теме
Python. Замена значений в np.array
ferzmikk,
Код: python
1.
2.
3.
4.
import numpy as np
x=np.array([1,2,3,4,5,6,7,8,9,10])
y={10:6,6:7}
[y.get(z,z)for z in x]
...
Рейтинг: 0 / 0
| Ответить | Цитировать | Написать  
07.02.2021, 14:39
    #40042806
ferzmikk
ferzmikk
Участник
Скрыть профиль Поместить в игнор-лист Сообщения автора в теме
Python. Замена значений в np.array
vikkiv,

Интересное решение. Спасибо!
...
Рейтинг: 0 / 0
| Ответить | Цитировать | Написать  
Форумы / PHP, Perl, Python [игнор отключен] [закрыт для гостей] / Python. Замена значений в np.array / 3 сообщений из 3, страница 1 из 1
Найденые пользователи ...
Разблокировать пользователей ...
Читали тему (0):
Читали форум (0):
Пользователи онлайн (0):
x
x
Закрыть


Просмотр
0 / 0
Close
Debug Console [Select Text]