Here are the examples of the python api numpy.arange.astype taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.
145 Examples
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_duplicate_columns(self):
# GH20839
odict = compat.OrderedDict
df = DataFrame(odict([('a', list('abc')),
('b', list(range(1, 4))),
('c', np.arange(3, 6).astype('u1')),
('d', np.arange(4.0, 7.0, dtype='float64')),
('e', [True, False, True]),
('f', pd.date_range('now', periods=3).values)]))
df.columns = ['a', 'a', 'b', 'b', 'b', 'c']
expected = DataFrame({'a': list(range(1, 4)),
'b': np.arange(3, 6).astype('u1')})
result = df.select_dtypes(include=[np.number], exclude=['floating'])
astert_frame_equal(result, expected)
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_bad_arg_raises(self):
df = DataFrame({'a': list('abc'),
'g': list(u('abc')),
'b': list(range(1, 4)),
'c': np.arange(3, 6).astype('u1'),
'd': np.arange(4.0, 7.0, dtype='float64'),
'e': [True, False, True],
'f': pd.date_range('now', periods=3).values})
msg = 'data type.*not understood'
with pytest.raises(TypeError, match=msg):
df.select_dtypes(['blargy, blarg, blarg'])
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_not_an_attr_but_still_valid_dtype(self):
df = DataFrame({'a': list('abc'),
'b': list(range(1, 4)),
'c': np.arange(3, 6).astype('u1'),
'd': np.arange(4.0, 7.0, dtype='float64'),
'e': [True, False, True],
'f': pd.date_range('now', periods=3).values})
df['g'] = df.f.diff()
astert not hasattr(np, 'u8')
r = df.select_dtypes(include=['i8', 'O'], exclude=['timedelta'])
e = df[['a', 'b']]
astert_frame_equal(r, e)
r = df.select_dtypes(include=['i8', 'O', 'timedelta64[ns]'])
e = df[['a', 'b', 'g']]
astert_frame_equal(r, e)
3
View Complete Implementation : test_array.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_sum(self):
data = np.arange(10).astype(float)
out = SparseArray(data).sum()
astert out == 45.0
data[5] = np.nan
out = SparseArray(data, fill_value=2).sum()
astert out == 40.0
out = SparseArray(data, fill_value=np.nan).sum()
astert out == 40.0
3
View Complete Implementation : test_array.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_numpy_sum(self):
data = np.arange(10).astype(float)
out = np.sum(SparseArray(data))
astert out == 45.0
data[5] = np.nan
out = np.sum(SparseArray(data, fill_value=2))
astert out == 40.0
out = np.sum(SparseArray(data, fill_value=np.nan))
astert out == 40.0
msg = "the 'dtype' parameter is not supported"
with pytest.raises(ValueError, match=msg):
np.sum(SparseArray(data), dtype=np.int64)
msg = "the 'out' parameter is not supported"
with pytest.raises(ValueError, match=msg):
np.sum(SparseArray(data), out=out)
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_duplicate_columns(self):
# GH20839
odict = compat.OrderedDict
df = DataFrame(odict([('a', list('abc')),
('b', list(range(1, 4))),
('c', np.arange(3, 6).astype('u1')),
('d', np.arange(4.0, 7.0, dtype='float64')),
('e', [True, False, True]),
('f', pd.date_range('now', periods=3).values)]))
df.columns = ['a', 'a', 'b', 'b', 'b', 'c']
expected = DataFrame({'a': list(range(1, 4)),
'b': np.arange(3, 6).astype('u1')})
result = df.select_dtypes(include=[np.number], exclude=['floating'])
astert_frame_equal(result, expected)
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_bad_datetime64(self):
df = DataFrame({'a': list('abc'),
'b': list(range(1, 4)),
'c': np.arange(3, 6).astype('u1'),
'd': np.arange(4.0, 7.0, dtype='float64'),
'e': [True, False, True],
'f': pd.date_range('now', periods=3).values})
with pytest.raises(ValueError, match='.+ is too specific'):
df.select_dtypes(include=['datetime64[D]'])
with pytest.raises(ValueError, match='.+ is too specific'):
df.select_dtypes(exclude=['datetime64[as]'])
3
View Complete Implementation : test_dtypes.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_select_dtypes_include_exclude_using_scalars(self):
df = DataFrame({'a': list('abc'),
'b': list(range(1, 4)),
'c': np.arange(3, 6).astype('u1'),
'd': np.arange(4.0, 7.0, dtype='float64'),
'e': [True, False, True],
'f': pd.Categorical(list('abc')),
'g': pd.date_range('20130101', periods=3),
'h': pd.date_range('20130101', periods=3,
tz='US/Eastern'),
'i': pd.date_range('20130101', periods=3,
tz='CET'),
'j': pd.period_range('2013-01', periods=3,
freq='M'),
'k': pd.timedelta_range('1 day', periods=3)})
ri = df.select_dtypes(include=np.number, exclude='floating')
ei = df[['b', 'c', 'k']]
astert_frame_equal(ri, ei)
3
View Complete Implementation : test_numeric.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_get_indexer(self):
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target)
expected = np.array([0, -1, 1, 2, 3, 4,
-1, -1, -1, -1], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target, method='pad')
expected = np.array([0, 0, 1, 2, 3, 4,
4, 4, 4, 4], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target, method='backfill')
expected = np.array([0, 1, 1, 2, 3, 4,
-1, -1, -1, -1], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)
3
View Complete Implementation : test_numeric.py
Copyright Apache License 2.0
Author : Frank-qlu
Copyright Apache License 2.0
Author : Frank-qlu
def test_get_indexer(self):
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target)
expected = np.array([0, -1, 1, 2, 3, 4,
-1, -1, -1, -1], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target, method='pad')
expected = np.array([0, 0, 1, 2, 3, 4,
4, 4, 4, 4], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)
target = UInt64Index(np.arange(10).astype('uint64') * 5 + 2**63)
indexer = self.index.get_indexer(target, method='backfill')
expected = np.array([0, 1, 1, 2, 3, 4,
-1, -1, -1, -1], dtype=np.intp)
tm.astert_numpy_array_equal(indexer, expected)