主页 / 问题 / 1130588
Accepted
Кирилл Маликов
Asked:2020-05-24 17:30:54 +0000 UTC2020-05-24 17:30:54 +0000 UTC

Python从不同的数组中成对添加元素

  • 772

我知道python不会出错，但是得到了某种荒谬的算术，请看。我有两个数组m_和m__. 现在我想成对添加它们的元素并获得一个新列表。

这里我上传了带有数据的源文件https://yadi.sk/d/hVZw12w-gI38iA

在输出端，我想得到一个数组sum_m，它对数组中元素的值m_和m__.

这是代码

import pandas as pd
import collections

""" Чтение файла """

file = r"C:\si2.csv"
df = pd.read_csv(file, sep=";")

""" Добавление нового столбца в датафрейм (разница) """

df["<DELTA>"] = df["<CLOSE>"] - df["<OPEN>"]
print(df.head(5))

massiv = []
for i in df["<DELTA>"]:
    massiv.append(i)

print(massiv)
print(len(massiv))
print("----------------------------------------")

""" Создание массивов в помощь для поиска индексов смены свечей """

m_ = []
m__ = []

for i in massiv:
    if i >= 0:
        m_.append(1)
    else:
        m_.append(-1)

print(" Первый массив", m_)
print(len(m_))

m__ = m_
m__.insert(0, 0)
m__ = m__[0: len(massiv)]

print(" Второй массив", m__)
print(len(m__))
print("----------------------------------------")

""" Поиск индексов смены свечей """

sum_m = [m_ + m__ for m_, m__ in zip(m_, m__)]
print(sum_m)
print(len(sum_m))

这就是输出，某种难以理解的算术。 在此处输入图像描述

在这里，我将清楚地展示我想要从前 4 个元素的算法中得到什么

在此处输入图像描述

我认为答案就在表面的某个地方，但我固执地看不到它。

python

2 个回答

  1. Best Answer
    import numpy as np

df["m_"] = np.where(df["<DELTA>"] >= 0, 1, -1)
df["m__"] = df["m_"].shift().fillna(0)
res = df.eval("res = m_ + m__")

    结果：

    In [13]: res
Out[13]:
      <OPEN>  <HIGH>  <LOW>  <CLOSE>  <DELTA>  m_  m__  res
0      64950   65000  64831    64898      -52  -1  0.0 -1.0
1      64881   64898  64836    64870      -11  -1 -1.0 -2.0
2      64872   64872  64758    64788      -84  -1 -1.0 -2.0
3      64786   64835  64773    64830       44   1 -1.0  0.0
4      64840   64915  64798    64899       59   1  1.0  2.0
...      ...     ...    ...      ...      ...  ..  ...  ...
4082   71060   71130  71028    71088       28   1 -1.0  0.0
4083   71086   71148  71014    71039      -47  -1  1.0  0.0
4084   71039   71061  70879    70909     -130  -1 -1.0 -2.0
4085   70907   71064  70905    71014      107   1 -1.0  0.0
4086   71013   71104  70976    71097       84   1  1.0  2.0

[4087 rows x 8 columns]
    • 2

  2. m__ = m_ 此行错误

    m__ 没有分配 m_ 数组，而是简单地通过 for 再次运行它，一切都解决了

    • 1

相关问题