# -*- coding: utf-8 -*-
"""
@description:
@author:XuMing
"""
from __future__ import print_function
from __future__ import unicode_literals

from numpy import *

a = array([[0, 1, 2, 3], [4, 5, 6, 7]])
print(a)

for row in a:
    print(row)

# 所有元素的迭代器：
for i in a.flat:
    print(i)


# 转置：
print(a.T)
print(a)
print(a.shape)  # 数组形状 (m,n,o,...)

print(a.size)  # 数组元素数
a.resize((4, 2))
print(a)
print(a.shape)

# 去除长度为1的维度：
a = a.squeeze()
print(a.shape)

# 复制：
a = array([[0, 1, 2, 3], [4, 5, 6, 7]])
b = a.copy()
b[0][0] = -1
print(b)

# 填充：
b.fill(9)
print(b)

# 转化为列表：
print(a.tolist())

# 复数
# 实部：
b = array([1 + 2j, 3 + 4j, 5 + 6j])
c = b.real
print(c)

# 虚部：
d = b.imag
print(d)

# 共轭：
print(b.conj())

# 保存成文本：
a.dump("file.txt")

# 字符串：
a.dumps()

# 写入文件
a.tofile('foo.csv', sep=',', format="%s")

# 查找排序
# 非零元素的索引：
b = a.nonzero()
print(a, b)

# 排序：
b = array([3, 2, 7, 4, 1])
b.sort()
print(b)

# 排序的索引位置：
b = array([2, 3, 1])
print(b.argsort(axis=-1))  # array([2, 0, 1], dtype=int64)
# 将 b 插入 a 中的索引，使得 a 保持有序：
a = array([1, 3, 4, 6])
b = array([0, 2, 5])
print(a.searchsorted(b))

# 元素数学操作
# 限制在一定范围：
a = array([[4, 1, 3], [2, 1, 5]])
print(a.clip(0, 2))

# 近似：
a = array([1.344, 2.449, 2.558])
b = a.round(decimals=2)
print(b)  # [ 1.34  2.45  2.56]

# 是否全部非零：
print(a.all())

import os

os.remove('foo.csv')
os.remove('file.txt')